diff --git a/src/viewport-three/ViewportCanvas.tsx b/src/viewport-three/ViewportCanvas.tsx
index 542f4e6d..1a3c4f40 100644
--- a/src/viewport-three/ViewportCanvas.tsx
+++ b/src/viewport-three/ViewportCanvas.tsx
@@ -197,12 +197,12 @@ export function ViewportCanvas({
   }, [whiteboxSelectionMode]);
 
   useLayoutEffect(() => {
-    hostRef.current?.updateDocument(
-      sceneDocument,
-      selection,
-      activeSelectionId
-    );
-  }, [sceneDocument, selection, activeSelectionId]);
+    hostRef.current?.updateSelection(selection, activeSelectionId);
+  }, [selection, activeSelectionId]);
+
+  useLayoutEffect(() => {
+    hostRef.current?.updateDocument(sceneDocument);
+  }, [sceneDocument]);
 
   useLayoutEffect(() => {
     hostRef.current?.setViewMode(viewMode);