diff --git a/overcooked_simulator/main.py b/overcooked_simulator/main.py
index b783eaa3594e673246a5f9890e4a926386ceb679..47897d248160f83aef69f958c1da01b2623c1725 100644
--- a/overcooked_simulator/main.py
+++ b/overcooked_simulator/main.py
@@ -1,7 +1,7 @@
 from player import Player
 import sys
-from simulation_runner import Simulator
-from overcooked_environment import Environment
+from overcooked_simulator.simulation_runner import Simulator
+from overcooked_simulator.overcooked_environment import Environment
 
 def main():
     simulator = Simulator(Environment, 300)
diff --git a/overcooked_simulator/overcooked_environment.py b/overcooked_simulator/overcooked_environment.py
index 946ca30820522b9dcd02f3a2925dc71268ad0be2..71060740332ed8a2c041cfac8a6dd841c1edeb64 100644
--- a/overcooked_simulator/overcooked_environment.py
+++ b/overcooked_simulator/overcooked_environment.py
@@ -2,7 +2,7 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from player import Player
+    from overcooked_simulator.player import Player
 from pathlib import Path
 import numpy as np
 
diff --git a/overcooked_simulator/player.py b/overcooked_simulator/player.py
index ad39b4e29a98c2fb41731f3d8f0a3391d2aae8c7..42f3d26fff900cbbd059de18c68657da34fbe558 100644
--- a/overcooked_simulator/player.py
+++ b/overcooked_simulator/player.py
@@ -1,5 +1,5 @@
 import numpy as np
-from overcooked_environment import Counter
+from overcooked_simulator.overcooked_environment import Counter
 
 
 class Player:
diff --git a/overcooked_simulator/simulation_runner.py b/overcooked_simulator/simulation_runner.py
index 1808898d1e0b887924096b1f5b627c031e11b9a8..846a56bcbb2e243c1df54ae287104c8eedacb90a 100644
--- a/overcooked_simulator/simulation_runner.py
+++ b/overcooked_simulator/simulation_runner.py
@@ -1,7 +1,7 @@
 from threading import Thread
 import time
-from overcooked_environment import Environment, Action
-from player import Player
+from overcooked_simulator.overcooked_environment import Environment, Action
+from overcooked_simulator.player import Player
 
 
 class Simulator(Thread):