diff --git a/run.py b/run.py
index 40d808ab42e5856fefa7f51b1896ef69bb33fed2..626b8e38fb0c18000388c64a3d19b8f5bb96ffe1 100644
--- a/run.py
+++ b/run.py
@@ -10,6 +10,7 @@ from flatland.envs.rail_env import RailEnvActions
 from flatland.evaluators.client import FlatlandRemoteClient
 from flatland.evaluators.client import TimeoutException
 
+from reinforcement_learning.ppo.ppo_agent import PPOAgent
 from utils.dead_lock_avoidance_agent import DeadLockAvoidanceAgent
 from utils.deadlock_check import check_if_all_blocked
 from utils.fast_tree_obs import FastTreeObs