diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 1b7100a9..c90a0006 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -241,6 +241,13 @@ import {
   applyWhiteboxBevelToMaterial,
   shouldApplyWhiteboxBevel
 } from "../rendering/whitebox-bevel-material";
+import {
+  ALL_RENDER_LAYER_MASK,
+  applyRendererRenderCategory,
+  applyRendererRenderCategoryFromMaterial,
+  enableCameraRendererRenderCategories,
+  enableObjectForAllRendererRenderCategories
+} from "../rendering/render-layers";
 import {
   collectWaterContactPatches,
   createWaterMaterial