diff --git a/DDPG_solver/main.py b/DDPG_solver/main.py
index 79b87e7..0bd634a 100644
--- a/DDPG_solver/main.py
+++ b/DDPG_solver/main.py
@@ -1,4 +1,3 @@
-from env import PartitionMazeEnv
 from utils import str2bool, evaluate_policy
 from datetime import datetime
 from DDPG import DDPG_agent
@@ -9,7 +8,7 @@ import argparse
 import torch
 import sys
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-
+from env import PartitionMazeEnv
 
 '''Hyperparameter Setting'''
 parser = argparse.ArgumentParser()