diff --git a/tests/test_geometry_factory.py b/tests/test_geometry_factory.py
index 5b4f85b6..21668880 100644
--- a/tests/test_geometry_factory.py
+++ b/tests/test_geometry_factory.py
@@ -26,7 +26,7 @@ class TestGeometryFactory(TestCase):
   def _get_citygml(self, file):
     if self._city_gml is None:
       file_path = (self._example_path / file).resolve()
-      self._city_gml = GeometryFactory('citygml', file_path)._city_debug
+      self._city_gml = GeometryFactory('citygml', file_path).city
       self.assertIsNotNone(self._city_gml, 'city is none')
     return self._city_gml
 
@@ -198,7 +198,7 @@ class TestGeometryFactory(TestCase):
     :return:
     """
     file = (self._example_path / 'gml_17_12_2020.gml').resolve()
-    city = GeometryFactory('citygml', file)._city_debug
+    city = GeometryFactory('citygml', file).city
 
     counter = 0
     for building in city.buildings:
diff --git a/tests/test_imports.py b/tests/test_imports.py
index 60432cca..025c262c 100644
--- a/tests/test_imports.py
+++ b/tests/test_imports.py
@@ -25,7 +25,7 @@ class MyTestCase(TestCase):
   def _get_city(self):
     if self._city_obj is None:
       file_path = (self._example_path / 'kelowna.obj').resolve()
-      self._city_obj = GeometryFactory('obj', file_path)._city_debug
+      self._city_obj = GeometryFactory('obj', file_path).city
     return self._city_obj
 
   def test_import_obj(self):