diff --git a/src/akkudoktoreos/utils/visualize.py b/src/akkudoktoreos/utils/visualize.py
index fc684c9..51ccc63 100644
--- a/src/akkudoktoreos/utils/visualize.py
+++ b/src/akkudoktoreos/utils/visualize.py
@@ -454,7 +454,9 @@ def prepare_visualize(
         [
             np.full(
                 len(parameters.ems.gesamtlast) - start_hour,
-                parameters.ems.einspeiseverguetung_euro_pro_wh,
+                parameters.ems.einspeiseverguetung_euro_pro_wh[start_hour:]
+                if isinstance(parameters.ems.einspeiseverguetung_euro_pro_wh, list)
+                else parameters.ems.einspeiseverguetung_euro_pro_wh,
             )
         ],
         title="Remuneration",
diff --git a/tests/testdata/optimize_input_1.json b/tests/testdata/optimize_input_1.json
index 88000e6..f5cde44 100644
--- a/tests/testdata/optimize_input_1.json
+++ b/tests/testdata/optimize_input_1.json
@@ -1,7 +1,14 @@
 {
     "ems": {
         "preis_euro_pro_wh_akku": 0.0001,
-        "einspeiseverguetung_euro_pro_wh": 0.00007,
+        "einspeiseverguetung_euro_pro_wh": [
+          0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007,
+          0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007,
+          0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007,
+          0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007,
+          0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007, 0.00007,
+          0.00007, 0.00007, 0.00007
+        ],
         "gesamtlast": [
           676.71, 876.19, 527.13, 468.88, 531.38, 517.95, 483.15, 472.28, 1011.68, 995.00,
           1053.07, 1063.91, 1320.56, 1132.03, 1163.67, 1176.82, 1216.22, 1103.78, 1129.12,