diff --git a/src/bonsai/bonsai/bim/module/model/decorator.py b/src/bonsai/bonsai/bim/module/model/decorator.py
index 6c1304f0f4..003a701bbe 100644
--- a/src/bonsai/bonsai/bim/module/model/decorator.py
+++ b/src/bonsai/bonsai/bim/module/model/decorator.py
@@ -538,7 +538,7 @@ class PolylineDecorator:
         polyline_verts = [Vector((p.x, p.y, p.z)) for p in polyline_points]
 
         # Area
-        if measure_type == "AREA" and polyline_data.area:
+        if measure_type == "POLY_AREA" and polyline_data.area:
             if len(polyline_verts) < 3:
                 return
             center = sum(polyline_verts, Vector()) / len(polyline_verts)  # Center between all polyline points
@@ -556,7 +556,7 @@ class PolylineDecorator:
             blf.draw(self.font_id, text)
 
         # Length
-        if measure_type in {"POLYLINE", "AREA"}:
+        if measure_type in {"POLYLINE", "POLY_AREA"}:
             if len(polyline_verts) < 3:
                 return
             total_length_text_coords = view3d_utils.location_3d_to_region_2d(region, rv3d, polyline_verts[-1])
@@ -765,7 +765,7 @@ class PolylineDecorator:
         # Area highlight
         if polyline_data:
             area = polyline_data.area.split(" ")[0]
-            if area:
+            if polyline_data.measurement_type == "POLY_AREA" and area:
                 if float(area) > 0:
                     tris = self.calculate_polygon(polyline_verts)["tris"]
                     self.draw_batch("TRIS", polyline_verts, transparent_color(decorator_color_special), tris)
diff --git a/src/bonsai/bonsai/bim/module/project/decorator.py b/src/bonsai/bonsai/bim/module/project/decorator.py
index 7d30a203a0..7f002f3d6a 100644
--- a/src/bonsai/bonsai/bim/module/project/decorator.py
+++ b/src/bonsai/bonsai/bim/module/project/decorator.py
@@ -329,7 +329,7 @@ class MeasureDecorator:
 
             all_positions = []
             for i in range(len(polyline_points)):
-                if i < 1 and measure_type == "AREA":
+                if i < 1 and measure_type == "POLY_AREA":
                     continue
                 if i == 0:
                     continue
@@ -374,7 +374,7 @@ class MeasureDecorator:
             polyline_verts = [Vector((p.x, p.y, p.z)) for p in polyline_points]
 
             # Area
-            if measure_type == "AREA" and polyline_data.area:
+            if measure_type == "POLY_AREA" and polyline_data.area:
                 if len(polyline_verts) < 3:
                     continue
                 center = sum(polyline_verts, Vector()) / len(polyline_verts)  # Center between all polyline points
@@ -392,7 +392,7 @@ class MeasureDecorator:
                 blf.draw(self.font_id, text)
 
             # Length
-            if measure_type in {"POLYLINE", "AREA"}:
+            if measure_type in {"POLYLINE", "POLY_AREA"}:
                 if len(polyline_verts) < 3:
                     continue
                 total_length_text_coords = view3d_utils.location_3d_to_region_2d(region, rv3d, polyline_verts[-1])
@@ -455,7 +455,7 @@ class MeasureDecorator:
             # Area highlight
             if polyline_data:
                 area = polyline_data.area.split(" ")[0]
-                if area:
+                if polyline_data.measurement_type == "POLY_AREA" and area:
                     if float(area) > 0:
                         tris = self.calculate_polygon(polyline_verts)["tris"]
                         self.draw_batch("TRIS", polyline_verts, transparent_color(decorator_color_special), tris)