diff --git a/carculator_truck/inventory.py b/carculator_truck/inventory.py
index 4dd44b6..068bef1 100644
--- a/carculator_truck/inventory.py
+++ b/carculator_truck/inventory.py
@@ -5242,14 +5242,10 @@ def set_inputs_in_A_matrix(self, array):
             ],
             -self.number_of_cars :,
         ] = (
-                        (
-                                array[self.array_inputs["energy battery mass"], :]
-                                * (
-                                        1
-                                        + array[
-                                            self.array_inputs["battery lifetime replacements"]]
-                                )
-                        )
+            (
+                array[self.array_inputs["energy battery mass"], :]
+                * (1 + array[self.array_inputs["battery lifetime replacements"]])
+            )
             / array[self.array_inputs["lifetime kilometers"]]
             / (array[self.array_inputs["total cargo mass"]] / 1000)
         )
@@ -6756,13 +6752,8 @@ def set_inputs_in_A_matrix_for_export(self, array):
                 )
             ],
             -self.number_of_cars :,
-        ] = (
-            array[self.array_inputs["energy battery mass"], :]
-            * (
-                    1
-                    + array[
-                      self.array_inputs["battery lifetime replacements"]]
-            )
+        ] = array[self.array_inputs["energy battery mass"], :] * (
+            1 + array[self.array_inputs["battery lifetime replacements"]]
         )
 
         # Noise emissions