diff --git a/hub/unittests/test_db_factory.py b/hub/unittests/test_db_factory.py
index 7d072ebf..709fbfad 100644
--- a/hub/unittests/test_db_factory.py
+++ b/hub/unittests/test_db_factory.py
@@ -223,14 +223,16 @@ TestDBFactory
       control.database.add_simulation_results(
         cte.INSEL_MEB,
         json.dumps({cte.INSEL_MEB: [
-          {"monthly_cooling": monthly_cooling.to_json()},
-          {"yearly_cooling": yearly_cooling.to_json()},
-          {"monthly_heating": monthly_heating.to_json()},
-          {"yearly_heating": yearly_heating.to_json()},
           {"monthly_cooling_peak_load": monthly_cooling_peak_load.to_json()},
           {"yearly_cooling_peak_load": yearly_cooling_peak_load.to_json()},
           {"monthly_heating_peak_load": monthly_heating_peak_load.to_json()},
           {"yearly_heating_peak_load": yearly_heating_peak_load.to_json()},
+          {"monthly_electrical_peak_load": monthly_electrical_peak_load.to_json()},
+          {"yearly_electrical_peak_load": yearly_electrical_peak_load.to_json()},
+          {"monthly_cooling_demand": monthly_cooling.to_json()},
+          {"yearly_cooling_demand": yearly_cooling.to_json()},
+          {"monthly_heating_demand": monthly_heating.to_json()},
+          {"yearly_heating_demand": yearly_heating.to_json()},
           {"monthly_lighting_electrical_demand": monthly_lighting_electrical_demand.to_json()},
           {"yearly_lighting_electrical_demand": yearly_lighting_electrical_demand.to_json()},
           {"monthly_appliances_electrical_demand": monthly_appliances_electrical_demand.to_json()},