diff --git a/hydromt_fiat/api/hydromt_fiat_vm.py b/hydromt_fiat/api/hydromt_fiat_vm.py
index 63e5daa2..a4b3c519 100644
--- a/hydromt_fiat/api/hydromt_fiat_vm.py
+++ b/hydromt_fiat/api/hydromt_fiat_vm.py
@@ -236,7 +236,8 @@ def new_additional_attributes(self, config_yaml):
 
     def new_ground_elevation(self, config_yaml):
         source = config_yaml.model_extra["update_ground_elevation"].source
-        self.fiat_model.exposure.setup_ground_elevation(source)
+        unit = config_yaml.model_extra["update_ground_elevation"].unit
+        self.fiat_model.exposure.setup_ground_elevation(source, unit)
 
     def new_max_potential_damages(self, config_yaml):
         source = config_yaml.model_extra["update_max_potential_damage"].source