diff --git a/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py b/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py
index d1fe477a..544d6ca3 100644
--- a/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py
+++ b/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py
@@ -173,11 +173,12 @@ def test_waning_compartments(self, *mocks):
 
             for cell in pop.cells:
                 cell_data = cell.compartment_counter.retrieve()
-                for status in [InfectionStatus.Recovered, InfectionStatus.Dead]:
+                for status in [InfectionStatus.Recovered,
+                               InfectionStatus.Dead]:
                     recov_dead_state_count += cell_data[status]
 
             self.assertNotEqual(np.sum(recov_dead_state_count),
-                             self.pop_params["population_size"])
+                                self.pop_params["population_size"])
 
     def test_no_infection(self, *mocks):
         """Basic functional test to ensure noone is infected when there are