diff --git a/mlir/utils/vscode/src/mlirContext.ts b/mlir/utils/vscode/src/mlirContext.ts
--- a/mlir/utils/vscode/src/mlirContext.ts
+++ b/mlir/utils/vscode/src/mlirContext.ts
@@ -251,10 +251,11 @@
     let middleware = {};
     if (!workspaceFolder) {
       middleware = {
-        didOpen : (document, next) => {
+        didOpen : (document, next) : Promise<void> => {
           if (!vscode.workspace.getWorkspaceFolder(document.uri)) {
-            next(document);
+            return next(document);
           }
+          return Promise.resolve();
         }
       };
     }