diff --git a/tests/test_cost_tracker.py b/tests/test_cost_tracker.py
index d784b26..9946ec3 100644
--- a/tests/test_cost_tracker.py
+++ b/tests/test_cost_tracker.py
@@ -1,4 +1,4 @@
-import cost_tracker
+from src import cost_tracker
 
 def test_estimate_cost():
     # Test unknown model
diff --git a/tests/test_gui2_layout.py b/tests/test_gui2_layout.py
index 608ad6e..36d7b23 100644
--- a/tests/test_gui2_layout.py
+++ b/tests/test_gui2_layout.py
@@ -1,4 +1,4 @@
-from gui_2 import App
+from src.gui_2 import App
 
 def test_gui2_hubs_exist_in_show_windows(app_instance: App) -> None:
  """
diff --git a/tests/test_gui_updates.py b/tests/test_gui_updates.py
index 2d773ae..379b186 100644
--- a/tests/test_gui_updates.py
+++ b/tests/test_gui_updates.py
@@ -7,7 +7,7 @@ from typing import Any
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
-from gui_2 import App
+from src.gui_2 import App
 
 def test_telemetry_data_updates_correctly(app_instance: Any) -> None:
  """
diff --git a/tests/test_layout_reorganization.py b/tests/test_layout_reorganization.py
index 26e7165..4ee4665 100644
--- a/tests/test_layout_reorganization.py
+++ b/tests/test_layout_reorganization.py
@@ -7,7 +7,7 @@ import os
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
-from gui_2 import App
+from src.gui_2 import App
 
 def test_new_hubs_defined_in_show_windows(mock_app: App) -> None:
  """
@@ -40,7 +40,7 @@ def test_old_windows_removed_from_gui2(app_instance_simple: Any) -> None:
 @pytest.fixture
 def app_instance_simple() -> Any:
  from unittest.mock import patch
- from gui_2 import App
+ from src.gui_2 import App
  with patch('src.models.load_config', return_value={'ai': {}, 'projects': {}, 'gui': {'show_windows': {}}}), \
    patch('src.app_controller.AppController._init_ai_and_hooks'), \
    patch('src.app_controller.AppController._fetch_models'), \
diff --git a/tests/test_log_management_ui.py b/tests/test_log_management_ui.py
index 1f4a224..2951a13 100644
--- a/tests/test_log_management_ui.py
+++ b/tests/test_log_management_ui.py
@@ -3,7 +3,7 @@ from unittest.mock import MagicMock, patch
 from pathlib import Path
 
 # We can safely import gui_2 if we don't instantiate App without mocking its threads
-from gui_2 import App
+from src.gui_2 import App
 
 @pytest.fixture
 def mock_config(tmp_path: Path) -> Path:
diff --git a/tests/test_mma_approval_indicators.py b/tests/test_mma_approval_indicators.py
index ab3f02b..24a2cdd 100644
--- a/tests/test_mma_approval_indicators.py
+++ b/tests/test_mma_approval_indicators.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 from unittest.mock import patch, MagicMock
 
-from gui_2 import App
+from src.gui_2 import App
 
 
 def _make_app(**kwargs):
diff --git a/tests/test_mma_dashboard_refresh.py b/tests/test_mma_dashboard_refresh.py
index 064c10d..ae21865 100644
--- a/tests/test_mma_dashboard_refresh.py
+++ b/tests/test_mma_dashboard_refresh.py
@@ -1,7 +1,7 @@
 import pytest
 from unittest.mock import patch, MagicMock
 from typing import Any
-from gui_2 import App
+from src.gui_2 import App
 
 
 @pytest.fixture
diff --git a/tests/test_mma_dashboard_streams.py b/tests/test_mma_dashboard_streams.py
index 4da98b5..ae7ee15 100644
--- a/tests/test_mma_dashboard_streams.py
+++ b/tests/test_mma_dashboard_streams.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 from unittest.mock import patch, MagicMock
 
-from gui_2 import App
+from src.gui_2 import App
 
 
 def _make_app(**kwargs):
diff --git a/tests/test_mma_orchestration_gui.py b/tests/test_mma_orchestration_gui.py
index 66df561..4b08ce3 100644
--- a/tests/test_mma_orchestration_gui.py
+++ b/tests/test_mma_orchestration_gui.py
@@ -1,7 +1,7 @@
 import json
 from unittest.mock import patch
 import time
-from gui_2 import App
+from src.gui_2 import App
 
 def test_mma_ui_state_initialization(app_instance: App) -> None:
  """Verifies that the new MMA UI state variables are initialized correctly."""
diff --git a/tests/test_mma_prompts.py b/tests/test_mma_prompts.py
index 8a9738f..f647459 100644
--- a/tests/test_mma_prompts.py
+++ b/tests/test_mma_prompts.py
@@ -1,4 +1,4 @@
-from mma_prompts import PROMPTS
+from src.mma_prompts import PROMPTS
 
 def test_tier1_epic_init_constraints() -> None:
  prompt = PROMPTS["tier1_epic_init"]