diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 1a4c93ed..bf083657 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -128,12 +128,15 @@ import {
 } from "../entities/entity-instances";
 import {
   cloneBrushGeometry,
+  createConeBrush,
   createRadialPrismBrush,
+  createTorusBrush,
   createWedgeBrush,
   deriveBrushSizeFromGeometry,
   scaleBrushGeometryToSize,
   updateBrush,
   DEFAULT_BOX_BRUSH_SIZE,
+  DEFAULT_TORUS_BRUSH_SIZE,
   type Brush,
   type BrushGeometry,
   type BoxBrush,