diff --git a/src/commands/compile.ts b/src/commands/compile.ts
index 10cf6a95..7364b852 100644
--- a/src/commands/compile.ts
+++ b/src/commands/compile.ts
@@ -138,7 +138,7 @@ export async function importFile(
         undefined,
         true
       );
-      documentContentProvider.update(serverUri.with({ scheme: OBJECTSCRIPT_FILE_SCHEMA }));
+      if (serverUri) documentContentProvider.update(serverUri);
     })
     .catch((error) => {
       if (error?.statusCode == 409) {
diff --git a/src/providers/DocumentContentProvider.ts b/src/providers/DocumentContentProvider.ts
index c9eae4a4..3298681e 100644
--- a/src/providers/DocumentContentProvider.ts
+++ b/src/providers/DocumentContentProvider.ts
@@ -257,7 +257,7 @@ export class DocumentContentProvider implements vscode.TextDocumentContentProvid
     }
   }
 
-  public update(uri: vscode.Uri, message?: string): void {
+  public update(uri: vscode.Uri): void {
     this.onDidChangeEvent.fire(uri);
   }
 }