diff --git a/src/utils.py b/src/utils.py
index cbba15e..394513f 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -72,7 +72,7 @@ def ann_to_xml(project_info, image_info, img_filename, result_ann_dir, ann):
 
     is_segmented = 0
     for label in ann.labels:
-        if label.obj_class.geometry != sly.Rectangle:
+        if label.obj_class.geometry_type != sly.Rectangle:
             is_segmented = 1
             break