diff --git a/tests/testdata/optimize_input_2.json b/tests/testdata/optimize_input_2.json
index d12cebb..b7cb777 100644
--- a/tests/testdata/optimize_input_2.json
+++ b/tests/testdata/optimize_input_2.json
@@ -160,7 +160,9 @@
         "min_soc_percentage": 0
     },
     "inverter": {
-        "max_power_wh": 10000
+        "device_id": "inverter1",
+        "max_power_wh": 10000,
+        "battery_id": "battery1"
     },
     "eauto": {
         "device_id": "ev1",
@@ -170,11 +172,6 @@
         "initial_soc_percentage": 5,
         "min_soc_percentage": 80
     },
-    "inverter": {
-        "device_id": "inverter1",
-        "max_power_wh": 10000,
-        "battery_id": "battery1"
-    },
     "dishwasher": {
         "device_id": "dishwasher1",
         "consumption_wh": 5000,