diff --git a/apps/vscode-e2e/src/suite/tools/apply-diff-native.test.ts b/apps/vscode-e2e/src/suite/tools/apply-diff-native.test.ts
index 3d453bdf89..8419b30dec 100644
--- a/apps/vscode-e2e/src/suite/tools/apply-diff-native.test.ts
+++ b/apps/vscode-e2e/src/suite/tools/apply-diff-native.test.ts
@@ -429,7 +429,7 @@ function validateInput(input) {
 					alwaysAllowReadOnlyOutsideWorkspace: true,
 					toolProtocol: "native", // Enable native tool calling
 					apiProvider: "openrouter", // Use OpenRouter provider
-					apiModelId: "openai/gpt-5.1", // GPT-5.1 supports native tools
+					openRouterModelId: "openai/gpt-5.1", // GPT-5.1 supports native tools
 				},
 				text: `Use apply_diff on the file ${testFile.name} to change "Hello World" to "Hello Universe". The file already exists with this content:
 ${testFile.content}
diff --git a/apps/vscode-e2e/src/suite/tools/apply-diff.test.ts b/apps/vscode-e2e/src/suite/tools/apply-diff.test.ts
index 499fe7ebb4..56c00d3155 100644
--- a/apps/vscode-e2e/src/suite/tools/apply-diff.test.ts
+++ b/apps/vscode-e2e/src/suite/tools/apply-diff.test.ts
@@ -234,6 +234,8 @@ function validateInput(input) {
 					alwaysAllowWrite: true,
 					alwaysAllowReadOnly: true,
 					alwaysAllowReadOnlyOutsideWorkspace: true,
+					apiProvider: "openrouter", // Use OpenRouter provider
+					openRouterModelId: "openai/gpt-5.1", // GPT-5.1 supports native tools
 				},
 				text: `Use apply_diff on the file ${testFile.name} to change "Hello World" to "Hello Universe". The file already exists with this content:
 ${testFile.content}\nAssume the file exists and you can modify it directly.`,