diff --git a/extensions/vscode-notebook-tests/src/extension.ts b/extensions/vscode-notebook-tests/src/extension.ts
index d555df6ca1f..e66444a70ac 100644
--- a/extensions/vscode-notebook-tests/src/extension.ts
+++ b/extensions/vscode-notebook-tests/src/extension.ts
@@ -71,7 +71,7 @@ export function activate(context: vscode.ExtensionContext): any {
 			task.replaceOutput([new vscode.NotebookCellOutput([
 				vscode.NotebookCellOutputItem.text('test output', 'text/html')
 			])]);
-			task.end({ success: true });
+			task.end(true);
 		}
 	};
 
diff --git a/src/vs/workbench/test/browser/api/extHostNotebookKernel2.test.ts b/src/vs/workbench/test/browser/api/extHostNotebookKernel2.test.ts
index bdbc2d94fa7..2ecbf92f398 100644
--- a/src/vs/workbench/test/browser/api/extHostNotebookKernel2.test.ts
+++ b/src/vs/workbench/test/browser/api/extHostNotebookKernel2.test.ts
@@ -173,7 +173,7 @@ suite('NotebookKernel', function () {
 		const cell1 = notebook.apiNotebook.cellAt(0);
 		const task = kernel.createNotebookCellExecution(cell1);
 		task.start();
-		task.end();
+		task.end(undefined);
 	});
 
 	test('createNotebookCellExecution, must be selected/associated', function () {