diff --git a/hydromt_fiat/api/hydromt_fiat_vm.py b/hydromt_fiat/api/hydromt_fiat_vm.py
index a009abfe..381840e0 100644
--- a/hydromt_fiat/api/hydromt_fiat_vm.py
+++ b/hydromt_fiat/api/hydromt_fiat_vm.py
@@ -212,7 +212,7 @@ def new_ground_floor_height(self, config_yaml):
         source = config_yaml.model_extra["update_ground_floor_height"].source
         gfh_attribute_name = config_yaml.model_extra[
             "update_ground_floor_height"
-        ].attribute_name
+        ].gfh_attribute_name
         gfh_method = config_yaml.model_extra["update_ground_floor_height"].gfh_method
         max_dist = config_yaml.model_extra["update_ground_floor_height"].max_dist
         self.fiat_model.exposure.setup_ground_floor_height(