diff --git a/hub/helpers/data/montreal_generation_system_to_hub_energy_generation_system.py b/hub/helpers/data/montreal_generation_system_to_hub_energy_generation_system.py
index 776ccd65..7a42c832 100644
--- a/hub/helpers/data/montreal_generation_system_to_hub_energy_generation_system.py
+++ b/hub/helpers/data/montreal_generation_system_to_hub_energy_generation_system.py
@@ -20,7 +20,7 @@ class MontrealGenerationSystemToHubEnergyGenerationSystem:
       'electricity generator': cte.ELECTRICITY_GENERATOR,
       'PV system': cte.PHOTOVOLTAIC,
       'heat pump': cte.HEAT_PUMP
-                        }
+    }
 
   @property
   def dictionary(self) -> dict:
diff --git a/hub/imports/energy_systems/montreal_custom_energy_system_parameters.py b/hub/imports/energy_systems/montreal_custom_energy_system_parameters.py
index a5ff316b..348df9fe 100644
--- a/hub/imports/energy_systems/montreal_custom_energy_system_parameters.py
+++ b/hub/imports/energy_systems/montreal_custom_energy_system_parameters.py
@@ -83,7 +83,7 @@ class MontrealCustomEnergySystemParameters:
   def _create_generation_systems(archetype_system):
     _generation_systems = []
     for archetype_generation_system in archetype_system.generation_systems:
-      if archetype_generation_system.system_type == 'PV system':
+      if archetype_generation_system.system_type == 'Photovoltaic':
         _generation_system = PvGenerationSystem()
         _type = 'PV system'
         _generation_system.system_type = Dictionaries().montreal_generation_system_to_hub_energy_generation_system[_type]