diff --git a/examples/atari/atari_qrdqn.py b/examples/atari/atari_qrdqn.py
index 06f1cbb..72915ef 100644
--- a/examples/atari/atari_qrdqn.py
+++ b/examples/atari/atari_qrdqn.py
@@ -7,12 +7,12 @@ import numpy as np
 import torch
 from atari_network import QRDQN
 from atari_wrapper import make_atari_env
-from torch.utils.tensorboard import SummaryWriter, WandbLogger
+from torch.utils.tensorboard import SummaryWriter
 
 from tianshou.data import Collector, VectorReplayBuffer
 from tianshou.policy import QRDQNPolicy
 from tianshou.trainer import offpolicy_trainer
-from tianshou.utils import TensorboardLogger
+from tianshou.utils import TensorboardLogger, WandbLogger
 
 
 def get_args():
diff --git a/examples/atari/atari_rainbow.py b/examples/atari/atari_rainbow.py
index 5109a1e..9b25450 100644
--- a/examples/atari/atari_rainbow.py
+++ b/examples/atari/atari_rainbow.py
@@ -7,12 +7,12 @@ import numpy as np
 import torch
 from atari_network import Rainbow
 from atari_wrapper import make_atari_env
-from torch.utils.tensorboard import SummaryWriter, WandbLogger
+from torch.utils.tensorboard import SummaryWriter
 
 from tianshou.data import Collector, PrioritizedVectorReplayBuffer, VectorReplayBuffer
 from tianshou.policy import RainbowPolicy
 from tianshou.trainer import offpolicy_trainer
-from tianshou.utils import TensorboardLogger
+from tianshou.utils import TensorboardLogger, WandbLogger
 
 
 def get_args():