diff --git a/src/bonsai/bonsai/bim/module/model/decorator.py b/src/bonsai/bonsai/bim/module/model/decorator.py
index 75df308aa7..3fa8acc0b6 100644
--- a/src/bonsai/bonsai/bim/module/model/decorator.py
+++ b/src/bonsai/bonsai/bim/module/model/decorator.py
@@ -492,10 +492,11 @@ class PolylineDecorator:
         new_line = 20
         for i, (key, field_name) in enumerate(texts.items()):
 
-            if self.tool_state and key != self.tool_state.input_type:
-                formatted_value = self.input_ui.get_formatted_value(key)
-            else:
-                formatted_value = self.input_ui.get_text_value(key)
+            if self.input_ui:
+                if self.tool_state and key != self.tool_state.input_type:
+                    formatted_value = self.input_ui.get_formatted_value(key)
+                else:
+                    formatted_value = self.input_ui.get_text_value(key)
 
             if formatted_value is None:
                 continue