diff --git a/src/composables/use-canvas-input.ts b/src/composables/use-canvas-input.ts
index acd197279..703358a49 100644
--- a/src/composables/use-canvas-input.ts
+++ b/src/composables/use-canvas-input.ts
@@ -76,6 +76,8 @@ const TOOL_TO_NODE: Partial<Record<Tool, NodeType>> = {
   RECTANGLE: 'RECTANGLE',
   ELLIPSE: 'ELLIPSE',
   LINE: 'LINE',
+  POLYGON: 'POLYGON',
+  STAR: 'STAR',
   TEXT: 'TEXT'
 }