diff --git a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt
index b2d4c577..81d97ba3 100644
--- a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt
+++ b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt
@@ -115,12 +115,21 @@ class ReviewLoopWorkflowTest {
 
         // implement claims/submits and may create a task, alongside its file write.
         assertEquals(
-            setOf("file_write", "task_create", "task_update", "task_context", "task_search"),
+            setOf(
+                "file_write", "task_create", "task_update", "task_context", "task_search",
+                "mcp__codebase-memory__search_code",
+                "mcp__codebase-memory__get_code_snippet",
+                "mcp__codebase-memory__trace_path",
+            ),
             graph.stages[StageId("implement")]!!.allowedTools,
         )
         // review reads the task and completes it on an approved verdict.
         assertEquals(
-            setOf("task_context", "task_update"),
+            setOf(
+                "task_context", "task_update",
+                "mcp__codebase-memory__search_code",
+                "mcp__codebase-memory__trace_path",
+            ),
             graph.stages[StageId("review")]!!.allowedTools,
         )
     }
diff --git a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt
index 684f4744..ab048c13 100644
--- a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt
+++ b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt
@@ -114,7 +114,12 @@ class RolePipelineWorkflowTest {
         assertTrue(implTools.containsAll(setOf("file_write", "file_edit", "task_update")))
         // reviewer reads the task and completes it on an approved verdict.
         assertEquals(
-            setOf("task_context", "task_update"),
+            setOf(
+                "task_context", "task_update",
+                "mcp__codebase-memory__search_code",
+                "mcp__codebase-memory__trace_path",
+                "mcp__codebase-memory__get_code_snippet",
+            ),
             graph.stages[StageId("reviewer")]!!.allowedTools,
         )
     }