diff --git a/env_summary_logger.py b/envs/env_summary_logger.py
similarity index 100%
rename from env_summary_logger.py
rename to envs/env_summary_logger.py
diff --git a/train.py b/train.py
index db1451c..5e0304f 100755
--- a/train.py
+++ b/train.py
@@ -1,11 +1,12 @@
 import time
 
 import numpy as np
+from tqdm import tqdm
+
 from base_train import BaseTrainer
-from env_summary_logger import EnvSummaryLogger
+from envs.env_summary_logger import EnvSummaryLogger
 from utils.lr_decay import LearningRateDecay
 from utils.utils import create_list_dirs
-from tqdm import tqdm
 
 
 class Trainer(BaseTrainer):