diff --git a/config.toml b/config.toml
index 8c1a239..efe7da0 100644
--- a/config.toml
+++ b/config.toml
@@ -1,6 +1,6 @@
 [ai]
-provider = "gemini_cli"
-model = "gemini-2.0-flash"
+provider = "gemini"
+model = "gemini-2.5-flash-lite"
 temperature = 0.0
 max_tokens = 8192
 history_trunc_limit = 8000
diff --git a/project_history.toml b/project_history.toml
index c55280c..516d166 100644
--- a/project_history.toml
+++ b/project_history.toml
@@ -8,5 +8,5 @@ active = "main"
 
 [discussions.main]
 git_commit = ""
-last_updated = "2026-03-05T14:42:31"
+last_updated = "2026-03-05T16:40:05"
 history = []
diff --git a/tests/test_log_registry.py b/tests/test_log_registry.py
index b5450bf..90fbaa0 100644
--- a/tests/test_log_registry.py
+++ b/tests/test_log_registry.py
@@ -4,7 +4,7 @@ import os
 from datetime import datetime, timedelta
 
 # This import is expected to fail with ImportError until log_registry.py is created.
-from log_registry import LogRegistry
+from src.log_registry import LogRegistry
 
 class TestLogRegistry(unittest.TestCase):
 
diff --git a/tests/test_logging_e2e.py b/tests/test_logging_e2e.py
index a77ac25..a1d8787 100644
--- a/tests/test_logging_e2e.py
+++ b/tests/test_logging_e2e.py
@@ -3,7 +3,7 @@ from typing import Any
 from pathlib import Path
 from datetime import datetime, timedelta
 import session_logger
-from log_registry import LogRegistry
+from src.log_registry import LogRegistry
 from log_pruner import LogPruner
 
 @pytest.fixture
diff --git a/tests/test_orchestrator_pm.py b/tests/test_orchestrator_pm.py
index be84221..ffb4a29 100644
--- a/tests/test_orchestrator_pm.py
+++ b/tests/test_orchestrator_pm.py
@@ -2,8 +2,8 @@ import unittest
 from typing import Any
 from unittest.mock import patch
 import json
-import orchestrator_pm
-import mma_prompts
+from src import orchestrator_pm
+from src import mma_prompts
 
 class TestOrchestratorPM(unittest.TestCase):
 
diff --git a/tests/test_orchestrator_pm_history.py b/tests/test_orchestrator_pm_history.py
index 1a12299..e522297 100644
--- a/tests/test_orchestrator_pm_history.py
+++ b/tests/test_orchestrator_pm_history.py
@@ -3,7 +3,7 @@ from unittest.mock import patch, MagicMock
 import shutil
 import json
 from pathlib import Path
-import orchestrator_pm
+from src import orchestrator_pm
 
 class TestOrchestratorPMHistory(unittest.TestCase):
  def setUp(self) -> None:
diff --git a/tests/test_track_state_persistence.py b/tests/test_track_state_persistence.py
index 0fb136e..242ebfb 100644
--- a/tests/test_track_state_persistence.py
+++ b/tests/test_track_state_persistence.py
@@ -3,7 +3,7 @@ from datetime import datetime
 # Import the real models
 from src.models import TrackState, Metadata, Ticket
 # Import the persistence functions from project_manager
-from project_manager import save_track_state, load_track_state
+from src.project_manager import save_track_state, load_track_state
 
 def test_track_state_persistence(tmp_path) -> None:
  """