From bd5136353274c484b22a6780b1d19df7384205d7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bruno=20Perdig=C3=A3o?= <bruno@brunopo.com>
Date: Thu, 12 Sep 2024 17:20:52 -0300
Subject: [PATCH] Small fix related to trying to use polyline wall without wall
 type.

---
 src/bonsai/bonsai/bim/module/model/wall.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/bonsai/bonsai/bim/module/model/wall.py b/src/bonsai/bonsai/bim/module/model/wall.py
index d881a52db3..e91a9d1216 100644
--- a/src/bonsai/bonsai/bim/module/model/wall.py
+++ b/src/bonsai/bonsai/bim/module/model/wall.py
@@ -317,6 +317,8 @@ class DrawPolylineWall(bpy.types.Operator, PolylineOperator):
     def modal(self, context, event):
         if not self.relating_type:
             self.report({"WARNING"}, "You need to select a wall type.")
+            PolylineDecorator.uninstall()
+            tool.Blender.update_viewport()
             return {"FINISHED"}
 
         PolylineDecorator.update(event, self.tool_state, self.input_ui, self.snapping_points[0])