diff --git a/backend/export/tests/test_export.py b/backend/export/tests/test_export.py
index b00d1744..f13ef374 100644
--- a/backend/export/tests/test_export.py
+++ b/backend/export/tests/test_export.py
@@ -284,8 +284,8 @@ def test_default_export_integration(db_session):
 
     assert df.shape == (
         10,
-        100,
-    ), "Expected dataframe shape to be (10, 100), got {}".format(df.shape)
+        101,
+    ), "Expected dataframe shape to be (10, 101), got {}".format(df.shape)
 
 
 def test_solar_with_battery_example(db_session):