diff --git a/hub/unittests/test_exports.py b/hub/unittests/test_exports.py
index 283c9595..7cf35a4c 100644
--- a/hub/unittests/test_exports.py
+++ b/hub/unittests/test_exports.py
@@ -71,8 +71,6 @@ class TestExports(TestCase):
       self._complete_city = self._get_complete_city(from_pickle)
       EnergyBuildingsExportsFactory(export_type, self._complete_city, self._output_path).export()
 
-
-
   def test_obj_export(self):
     """
     export to obj
@@ -102,13 +100,15 @@ class TestExports(TestCase):
     """
     export to IDF
     """
-    city = self._get_citygml('EV_GM_MB_LoD2.gml')
-    for building in city.buildings:
-      building.year_of_construction = 2006
-      if building.function is None:
-        building.function = 'large office'
-    ConstructionFactory('nrel', city).enrich()
-    UsageFactory('comnet', city).enrich()
+    file = 'FZK_Haus_LoD_2.gml'
+    file_path = (self._example_path / file).resolve()
+    city = GeometryFactory('citygml',
+                           path=file_path,
+                           function_to_hub=Dictionaries().alkis_function_to_hub_function).city
+    self.assertIsNotNone(city, 'city is none')
+    ConstructionFactory('nrcan', city=city)
+    ConstructionFactory('nrcan', city).enrich()
+    UsageFactory('nrcan', city).enrich()
     try:
       EnergyBuildingsExportsFactory('idf', city, self._output_path).export()
       EnergyBuildingsExportsFactory('idf', city, self._output_path,