diff --git a/tests/mock_gemini_cli.py b/tests/mock_gemini_cli.py
index 8545766..8646e59 100644
--- a/tests/mock_gemini_cli.py
+++ b/tests/mock_gemini_cli.py
@@ -42,10 +42,10 @@ def main() -> None:
   }), flush=True)
   return
 
- elif 'PATH: Sprint Planning' in prompt:
+ elif 'PATH: Sprint Planning' in prompt or 'generate the implementation tickets' in prompt:
   mock_response = [
-   {"id": "mock-ticket-1", "type": "Ticket", "goal": "Mock Ticket 1", "target_file": "file1.py", "depends_on": [], "context_requirements": "req 1"},
-   {"id": "mock-ticket-2", "type": "Ticket", "goal": "Mock Ticket 2", "target_file": "file2.py", "depends_on": ["mock-ticket-1"], "context_requirements": "req 2"}
+   {"id": "mock-ticket-1", "description": "Mock Ticket 1", "status": "todo", "assigned_to": "worker", "depends_on": []},
+   {"id": "mock-ticket-2", "description": "Mock Ticket 2", "status": "todo", "assigned_to": "worker", "depends_on": ["mock-ticket-1"]}
   ]
   print(json.dumps({
    "type": "message",