diff --git a/vscode-extension/server/src/server.ts b/vscode-extension/server/src/server.ts
index 61246bc..c265b28 100644
--- a/vscode-extension/server/src/server.ts
+++ b/vscode-extension/server/src/server.ts
@@ -63,6 +63,7 @@ function handleInitialize(): InitializeResult {
 
 function handleDocumentOpen(event: TextDocumentChangeEvent<TextDocument>) {
   const document = event.document
+  setGlobals(Object.keys(globals))
   const tree = parser.parse(document.getText())
   documentTrees.set(document.uri, tree)
 }
@@ -82,6 +83,7 @@ function handleDocumentChange(change: TextDocumentChangeEvent<TextDocument>) {
   const document = change.document
 
   // Parse and cache
+  setGlobals(Object.keys(globals))
   const tree = parser.parse(document.getText())
   documentTrees.set(document.uri, tree)