diff --git a/model_data/tests/test_lighting_attributes.py b/model_data/tests/test_lighting_attributes.py
index 64044abd..392e10d7 100644
--- a/model_data/tests/test_lighting_attributes.py
+++ b/model_data/tests/test_lighting_attributes.py
@@ -5,13 +5,11 @@ from model_data.epc_attributes.LightingAttributes import LightingAttributes
 
 # An example averages dataset to use in tests. It is a dictionary where the key is a lighting description and the
 # value is the expected proportion.
-averages = pd.DataFrame(
-    [
-        {"lighting-description": "good lighting efficiency", "low-energy-lighting": 0.75},
-        {"lighting-description": "excellent lighting efficiency", "low-energy-lighting": 1.0},
-        {"lighting-description": "below average lighting efficiency", "low-energy-lighting": 0.25}
-    ]
-)
+averages = [
+    {"lighting-description": "good lighting efficiency", "low-energy-lighting": 0.75},
+    {"lighting-description": "excellent lighting efficiency", "low-energy-lighting": 1.0},
+    {"lighting-description": "below average lighting efficiency", "low-energy-lighting": 0.25}
+]
 
 
 class TestLightingAttributes: