diff --git a/tests/models/test_short_term_storage_complex.py b/tests/models/test_short_term_storage_complex.py
index 5f85110f..0d23c050 100644
--- a/tests/models/test_short_term_storage_complex.py
+++ b/tests/models/test_short_term_storage_complex.py
@@ -1,6 +1,7 @@
-import pandas as pd
 import math
 
+import pandas as pd
+
 from andromede.libs.standard import (
     BALANCE_PORT_TYPE,
     DEMAND_MODEL,
@@ -27,7 +28,7 @@
 
 
 def generate_data(
-        efficiency: float, horizon: int, scenarios: int
+    efficiency: float, horizon: int, scenarios: int
 ) -> TimeScenarioSeriesData:
     # Create an empty DataFrame with index being the range of the horizon
     data = pd.DataFrame(index=range(horizon))