diff --git a/geojson_modelica_translator/model_connectors/model_base.py b/geojson_modelica_translator/model_connectors/model_base.py
index d085b130e..810904f75 100644
--- a/geojson_modelica_translator/model_connectors/model_base.py
+++ b/geojson_modelica_translator/model_connectors/model_base.py
@@ -62,7 +62,7 @@ def __init__(self, system_parameters, template_dir):
                 }
             # Get access to loop order output from ThermalNetwork package.
             if "fifth_generation" in district_params and "ghe_parameters" in district_params["fifth_generation"]:
-                self.loop_order: list = load_loop_order(self.system_parameters.filename)
+                self.loop_order = load_loop_order(self.system_parameters.filename)
 
     def ft2_to_m2(self, area_in_ft2: float) -> float:
         """Converts square feet to square meters
diff --git a/tests/model_connectors/test_district_multi_ghe.py b/tests/model_connectors/test_district_multi_ghe.py
index cdf21302b..44e4df35f 100644
--- a/tests/model_connectors/test_district_multi_ghe.py
+++ b/tests/model_connectors/test_district_multi_ghe.py
@@ -43,7 +43,7 @@ def setUp(self):
         sys_params = SystemParameters(sys_param_filename)
 
         # read the loop order and create building groups
-        loop_order: list = load_loop_order(sys_param_filename)
+        loop_order = load_loop_order(sys_param_filename)
 
         # create ambient water stub
         ambient_water_stub = NetworkDistributionPump(sys_params)
diff --git a/tests/model_connectors/test_district_single_ghe.py b/tests/model_connectors/test_district_single_ghe.py
index 21999e1b3..fffe4695b 100644
--- a/tests/model_connectors/test_district_single_ghe.py
+++ b/tests/model_connectors/test_district_single_ghe.py
@@ -36,7 +36,7 @@ def setUp(self):
         sys_params = SystemParameters(sys_param_filename)
 
         # read the loop order and create building groups
-        loop_order: list = load_loop_order(sys_param_filename)
+        loop_order = load_loop_order(sys_param_filename)
 
         # create ambient water loop stub
         ambient_water_stub = NetworkDistributionPump(sys_params)