diff --git a/vscode-dotty/src/worksheet.ts b/vscode-dotty/src/worksheet.ts
index 974a1a917ba6..a95a854a98f0 100644
--- a/vscode-dotty/src/worksheet.ts
+++ b/vscode-dotty/src/worksheet.ts
@@ -325,16 +325,11 @@ export class WorksheetProvider implements Disposable {
       codeLensProvider,
       vscode.languages.registerCodeLensProvider(documentSelector, codeLensProvider),
       vscode.workspace.onWillSaveTextDocument(event => {
+        const runWorksheetOnSave = vscode.workspace.getConfiguration("dotty").get("runWorksheetOnSave")
         const worksheet = this.worksheetFor(event.document)
         if (worksheet) {
           event.waitUntil(Promise.resolve(worksheet.prepareRun()))
-        }
-      }),
-      vscode.workspace.onDidSaveTextDocument(document => {
-        const runWorksheetOnSave = vscode.workspace.getConfiguration("dotty").get("runWorksheetOnSave")
-        const worksheet = this.worksheetFor(document)
-        if (runWorksheetOnSave && worksheet) {
-          worksheet.run()
+          if (runWorksheetOnSave) worksheet.run()
         }
       }),
       vscode.workspace.onDidCloseTextDocument(document => {