diff --git a/team_code/orion_b2d_agent.py b/team_code/orion_b2d_agent.py
index ce49090..9e13c1e 100644
--- a/team_code/orion_b2d_agent.py
+++ b/team_code/orion_b2d_agent.py
@@ -18,8 +18,8 @@
 import numpy as np
 from PIL import Image
 from torchvision import transforms as T
-from Bench2DriveZoo.team_code.pid_controller import PIDController
-from Bench2DriveZoo.team_code.planner import RoutePlanner
+from pid_controller import PIDController
+from planner import RoutePlanner
 from leaderboard.autoagents import autonomous_agent
 from mmcv import Config
 from mmcv.models import build_model