diff --git a/freqtrade/freqai/RL/TensorboardCallback.py b/freqtrade/freqai/RL/TensorboardCallback.py
index 282e60b0d..3924f9d2c 100644
--- a/freqtrade/freqai/RL/TensorboardCallback.py
+++ b/freqtrade/freqai/RL/TensorboardCallback.py
@@ -3,9 +3,8 @@ from typing import Any, Dict, Type, Union
 
 from stable_baselines3.common.callbacks import BaseCallback
 from stable_baselines3.common.logger import HParam
-from stable_baselines3.common.vec_env import DummyVecEnv, SubprocVecEnv, VecMonitor
 
-from freqtrade.freqai.RL.BaseEnvironment import BaseActions, BaseEnvironment
+from freqtrade.freqai.RL.BaseEnvironment import BaseActions
 
 
 class TensorboardCallback(BaseCallback):
@@ -17,8 +16,6 @@ class TensorboardCallback(BaseCallback):
         super().__init__(verbose)
         self.model: Any = None
         self.logger = None  # type: Any
-        self.training_env: Union[BaseEnvironment, SubprocVecEnv,
-                                 VecMonitor, DummyVecEnv] = DummyVecEnv()
         self.actions: Type[Enum] = actions
 
     def _on_training_start(self) -> None: