diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 1d69c2e9..371b60fb 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -128,6 +128,8 @@ import {
 } from "../entities/entity-instances";
 import {
   cloneBrushGeometry,
+  createRadialPrismBrush,
+  createWedgeBrush,
   deriveBrushSizeFromGeometry,
   scaleBrushGeometryToSize,
   updateBrush,