diff --git a/simulation/user_agent.py b/simulation/user_agent.py
index 107c00b..40a4231 100644
--- a/simulation/user_agent.py
+++ b/simulation/user_agent.py
@@ -1,7 +1,7 @@
 import time
 import random
 from typing import Any, Callable
-import ai_client
+from src import ai_client
 
 class UserSimAgent:
  def __init__(self, hook_client: Any, model: str = "gemini-2.5-flash-lite", enable_delays: bool = True) -> None:
diff --git a/tests/test_user_agent.py b/tests/test_user_agent.py
index 6c72d26..670b3d5 100644
--- a/tests/test_user_agent.py
+++ b/tests/test_user_agent.py
@@ -13,6 +13,7 @@ def test_user_agent_instantiation() -> None:
 
 def test_perform_action_with_delay() -> None:
  agent = UserSimAgent(hook_client=None)
+ agent.enable_delays = False
  called = False
 
  def action():
diff --git a/tests/test_visual_mma.py b/tests/test_visual_mma.py
index 886cb60..0e6ffd4 100644
--- a/tests/test_visual_mma.py
+++ b/tests/test_visual_mma.py
@@ -1,6 +1,9 @@
 import time
+import pytest
 from src import api_hook_client
 
+@pytest.mark.integration
+@pytest.mark.live
 def test_visual_mma_components(live_gui):
  """
     Refactored visual MMA verification using the live_gui fixture.
@@ -22,11 +25,11 @@ def test_visual_mma_components(live_gui):
   "description": "A track to verify MMA UI components"
  }
  tickets_data = [
-  {"id": "TICKET-001", "target_file": "core.py", "status": "todo"},
-  {"id": "TICKET-002", "target_file": "utils.py", "status": "running"},
-  {"id": "TICKET-003", "target_file": "tests.py", "status": "complete"},
-  {"id": "TICKET-004", "target_file": "api.py", "status": "blocked"},
-  {"id": "TICKET-005", "target_file": "gui.py", "status": "paused"},
+  {"id": "TICKET-001", "target_file": "core.py", "status": "todo", "description": "1", "assigned_to": "Worker"},
+  {"id": "TICKET-002", "target_file": "utils.py", "status": "running", "description": "2", "assigned_to": "Worker"},
+  {"id": "TICKET-003", "target_file": "tests.py", "status": "complete", "description": "3", "assigned_to": "Worker"},
+  {"id": "TICKET-004", "target_file": "api.py", "status": "blocked", "description": "4", "assigned_to": "Worker"},
+  {"id": "TICKET-005", "target_file": "gui.py", "status": "paused", "description": "5", "assigned_to": "Worker"},
  ]
 
  print("\nPushing MMA state update...")
diff --git a/tests/test_negative_flows.py b/tests/test_z_negative_flows.py
similarity index 100%
rename from tests/test_negative_flows.py
rename to tests/test_z_negative_flows.py