diff --git a/ogcore/TPI.py b/ogcore/TPI.py
index 1ae4873a4..15016fcd0 100644
--- a/ogcore/TPI.py
+++ b/ogcore/TPI.py
@@ -665,8 +665,12 @@ def run_TPI(p, client=None):
         D_f = np.zeros(p.T + p.S)
     else:
         if p.baseline_spending:
-            TR = np.concatenate((TRbaseline[:p.T], np.ones(p.S) * ss_vars["TR_ss"]))
-            G = np.concatenate((Gbaseline[:p.T], np.ones(p.S) * ss_vars["Gss"]))
+            TR = np.concatenate(
+                (TRbaseline[: p.T], np.ones(p.S) * ss_vars["TR_ss"])
+            )
+            G = np.concatenate(
+                (Gbaseline[: p.T], np.ones(p.S) * ss_vars["Gss"])
+            )
         else:
             TR = p.alpha_T * Y
             G = np.ones(p.T + p.S) * ss_vars["Gss"]