diff --git a/.changeset/fine-eels-find.md b/.changeset/fine-eels-find.md
new file mode 100644
index 0000000000..169eb8c12c
--- /dev/null
+++ b/.changeset/fine-eels-find.md
@@ -0,0 +1,5 @@
+---
+"roo-cline": patch
+---
+
+Improve the prompt for the "modes" e2e test
diff --git a/apps/vscode-e2e/src/suite/modes.test.ts b/apps/vscode-e2e/src/suite/modes.test.ts
index edc93d4c9d..817d5f71ce 100644
--- a/apps/vscode-e2e/src/suite/modes.test.ts
+++ b/apps/vscode-e2e/src/suite/modes.test.ts
@@ -1,40 +1,24 @@
 import * as assert from "assert"
 
-import type { ClineMessage } from "@roo-code/types"
-
 import { waitUntilCompleted } from "./utils"
 
 suite("Roo Code Modes", () => {
 	test("Should handle switching modes correctly", async () => {
-		const api = globalThis.api
-
-		const switchModesPrompt =
-			"For each mode (Architect, Ask, Debug) respond with the mode name and what it specializes in after switching to that mode."
-
-		const messages: ClineMessage[] = []
-		const modeSwitches: string[] = []
+		const modes: string[] = []
 
-		api.on("taskModeSwitched", (_taskId, mode) => {
-			console.log("taskModeSwitched", mode)
-			modeSwitches.push(mode)
-		})
-
-		api.on("message", ({ message }) => {
-			if (message.type === "say" && message.partial === false) {
-				messages.push(message)
-			}
-		})
+		globalThis.api.on("taskModeSwitched", (_taskId, mode) => modes.push(mode))
 
-		const switchModesTaskId = await api.startNewTask({
+		const switchModesTaskId = await globalThis.api.startNewTask({
 			configuration: { mode: "code", alwaysAllowModeSwitch: true, autoApprovalEnabled: true },
-			text: switchModesPrompt,
+			text: "For each of `architect`, `ask`, and `debug` use the `switch_mode` tool to switch to that mode.",
 		})
 
-		await waitUntilCompleted({ api, taskId: switchModesTaskId })
-		await api.cancelCurrentTask()
+		await waitUntilCompleted({ api: globalThis.api, taskId: switchModesTaskId })
+		await globalThis.api.cancelCurrentTask()
 
-		assert.ok(modeSwitches.includes("architect"))
-		assert.ok(modeSwitches.includes("ask"))
-		assert.ok(modeSwitches.includes("debug"))
+		assert.ok(modes.includes("architect"))
+		assert.ok(modes.includes("ask"))
+		assert.ok(modes.includes("debug"))
+		assert.ok(modes.length === 3)
 	})
 })