diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index 5764f865f712514e3d4cbef1648822dcc528554d..4f823be331850668e18bfdf66d35a915e3f6ccdc 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -6,7 +6,6 @@ from collections import deque
 import matplotlib.pyplot as plt
 import numpy as np
 import torch
-from dueling_double_dqn import Agent
 from importlib_resources import path
 
 import torch_training.Nets
@@ -15,8 +14,11 @@ from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.predictions import ShortestPathPredictorForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool
+from torch_training.dueling_double_dqn import Agent
 from utils.observation_utils import norm_obs_clip, split_tree
 
+print("multi_agent_trainging.py (1)")
+
 
 def main(argv):
     try:
@@ -219,6 +221,10 @@ def main(argv):
     plt.show()
 
 
+print("multi_agent_trainging.py (2)")
+
 if __name__ == '__main__':
     print("main")
     main(sys.argv[1:])
+
+print("multi_agent_trainging.py (3)")