diff --git a/src/bonsai/bonsai/bim/module/model/product.py b/src/bonsai/bonsai/bim/module/model/product.py
index eba74e4cf2..88a5ac53df 100644
--- a/src/bonsai/bonsai/bim/module/model/product.py
+++ b/src/bonsai/bonsai/bim/module/model/product.py
@@ -116,7 +116,7 @@ class AddOccurrence(bpy.types.Operator, PolylineOperator):
         if not self.relating_type:
             return {"FINISHED"}
 
-        result = tool.Polyline.insert_polyline_point(self.input_ui, tool_state)
+        result = tool.Polyline.insert_polyline_point(self.input_ui)
         if result:
             self.report({"WARNING"}, result)
 
@@ -139,7 +139,7 @@ class AddOccurrence(bpy.types.Operator, PolylineOperator):
 
         point = context.scene.BIMPolylineProperties.insertion_polyline[0].polyline_points[0]
         context.scene.cursor.location = Vector((point.x, point.y, point.z))
-        tool.Snap.clear_polyline()
+        tool.Polyline.clear_polyline()
 
         bpy.ops.bim.add_constr_type_instance("INVOKE_DEFAULT")
 
diff --git a/src/bonsai/bonsai/tool/polyline.py b/src/bonsai/bonsai/tool/polyline.py
index 33abb8bb72..8851f5c7c2 100644
--- a/src/bonsai/bonsai/tool/polyline.py
+++ b/src/bonsai/bonsai/tool/polyline.py
@@ -514,7 +514,7 @@ class Polyline(bonsai.core.tool.Polyline):
         return format_distance(value * factor, precision=precision, suppress_zero_inches=True, in_unit_length=True)
 
     @classmethod
-    def insert_polyline_point(cls, input_ui, tool_state):
+    def insert_polyline_point(cls, input_ui, tool_state=None):
         x = input_ui.get_number_value("X")
         y = input_ui.get_number_value("Y")
         if input_ui.get_number_value("Z") is not None:
@@ -525,7 +525,7 @@ class Polyline(bonsai.core.tool.Polyline):
         a = input_ui.get_formatted_value("A")
 
         snap_vertex = bpy.context.scene.BIMPolylineProperties.snap_mouse_point[0]
-        if tool_state.use_default_container:
+        if tool_state and tool_state.use_default_container:
             z = tool.Ifc.get_object(tool.Root.get_default_container()).location.z
 
         if x is None and y is None: