diff --git a/packages/kilo-vscode/tests/unit/tool-info-contract.test.ts b/packages/kilo-vscode/tests/unit/tool-info-contract.test.ts
index cceca3251f9..5f94f63285a 100644
--- a/packages/kilo-vscode/tests/unit/tool-info-contract.test.ts
+++ b/packages/kilo-vscode/tests/unit/tool-info-contract.test.ts
@@ -1,7 +1,7 @@
 /**
  * getToolInfo() return shape contract test.
  *
- * VscodeSessionTurn calls getToolInfo() from @kilocode/kilo-ui/message-part
+ * TaskToolExpanded calls getToolInfo() from @kilocode/kilo-ui/message-part
  * and relies on the returned ToolInfo having `icon` and `title` fields.
  *
  * This test verifies that the ToolInfo type exported from message-part.tsx
diff --git a/packages/kilo-vscode/tests/unit/tool-registry-contract.test.ts b/packages/kilo-vscode/tests/unit/tool-registry-contract.test.ts
index 9c34ad5aa11..ecadf08f35e 100644
--- a/packages/kilo-vscode/tests/unit/tool-registry-contract.test.ts
+++ b/packages/kilo-vscode/tests/unit/tool-registry-contract.test.ts
@@ -3,7 +3,7 @@
  *
  * The kilo-vscode webview overrides specific tool names in
  * VscodeToolOverrides.tsx (e.g. "bash") and TaskToolExpanded.tsx (e.g. "task").
- * VscodeSessionTurn also calls getToolInfo() from @kilocode/kilo-ui/message-part.
+ * VscodeSessionTurn imports Message/UserMessageDisplay from @kilocode/kilo-ui/message-part.
  *
  * This test verifies that the tool names we depend on are still registered
  * in the upstream ToolRegistry (message-part.tsx). If upstream removes or
@@ -22,7 +22,7 @@ const MESSAGE_PART_FILE = path.join(MONOREPO_ROOT, "packages/ui/src/components/m
  * Sources:
  *   - VscodeToolOverrides.tsx: "bash"
  *   - TaskToolExpanded.tsx:    "task"
- *   - VscodeSessionTurn.tsx uses getToolInfo() which handles all of these
+ *   - TaskToolExpanded.tsx uses getToolInfo() which handles all of these
  */
 const TOOL_NAMES_WE_DEPEND_ON = ["bash", "task", "read", "write", "glob", "edit", "todowrite"]