diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index d9d4dd60..13547594 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -210,6 +210,7 @@ export class ViewportHost {
   updateDocument(document: SceneDocument, selection: EditorSelection) {
     this.currentDocument = document;
     this.currentSelection = selection;
+    this.rebuildLocalLights(document);
     this.rebuildBrushMeshes(document, selection);
     this.rebuildEntityMarkers(document, selection);
     this.rebuildModelInstances(document, selection);