diff --git a/src/viewport-three/viewport-focus.ts b/src/viewport-three/viewport-focus.ts
index ab05032c..1e41cfa2 100644
--- a/src/viewport-three/viewport-focus.ts
+++ b/src/viewport-three/viewport-focus.ts
@@ -316,7 +316,7 @@ function createMultiSelectionFocusTarget(
         const entity = document.entities[entityId];
 
         if (entity !== undefined) {
-          includeEntity(bounds, entity);
+          includeEntity(bounds, entity, document);
         }
       }
       break;
@@ -526,7 +526,7 @@ export function resolveViewportDocumentBounds(
   }
 
   for (const entity of Object.values(document.entities)) {
-    includeEntity(bounds, entity);
+    includeEntity(bounds, entity, document);
   }
 
   return finishSceneBounds(bounds);
@@ -571,7 +571,7 @@ export function resolveViewportFocusTarget(document: SceneDocument, selection: E
     const entity = document.entities[selectedEntityId];
 
     if (entity !== undefined) {
-      return createEntityFocusTarget(entity);
+      return createEntityFocusTarget(entity, document);
     }
   }