diff --git a/hydromt_fiat/workflows/exposure_vector.py b/hydromt_fiat/workflows/exposure_vector.py
index 61e00fc3..cc9628f8 100644
--- a/hydromt_fiat/workflows/exposure_vector.py
+++ b/hydromt_fiat/workflows/exposure_vector.py
@@ -2012,7 +2012,7 @@ def set_height_relative_to_reference(
             # must be sorted on the object_id.
             identifier = (
                 "object_id"
-                if "object_id" in modified_objects_gdf.columns
+                if "Object ID" in modified_objects_gdf.columns
                 else "object_id"
             )