diff --git a/src/common/architecture.py b/src/common/architecture.py
index 9f57109..581ab11 100644
--- a/src/common/architecture.py
+++ b/src/common/architecture.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 
-from assignment.simulator.src.scheduler import Scheduler
+from src.common.scheduler import Scheduler
 
 @dataclass
 class Architecture:
diff --git a/src/common/budget.py b/src/common/budget.py
index 3e4286d..f5379ac 100644
--- a/src/common/budget.py
+++ b/src/common/budget.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 
-from assignment.simulator.src.scheduler import Scheduler
+from src.common.scheduler import Scheduler
 
 @dataclass
 class Budget: