diff --git a/src/lib/components/WaterfallView.svelte b/src/lib/components/WaterfallView.svelte
index 8155632d5d63897d1eb0266ef41cf5277c2f56ed..bc5f23c44b4d2ff81826f76a73b9c3ccccff26fc 100644
--- a/src/lib/components/WaterfallView.svelte
+++ b/src/lib/components/WaterfallView.svelte
@@ -189,14 +189,16 @@
     if (shared.savedSituation !== undefined) {
       situation = shared.savedSituation
       delete shared.savedSituation
-      if (shared.savedSituationIndex !== undefined)
+      if (shared.savedSituationIndex !== undefined) {
         shared.testCases[shared.savedSituationIndex] = situation
-      delete shared.savedSituationIndex
+        delete shared.savedSituationIndex
+      }
     }
   }
 
   function requestAxesCalculation() {
     shared.savedSituation = situation
+    shared.savedSituationIndex = situationIndex
     situation = structuredClone($state.snapshot(situation))
     shared.testCases[situationIndex] = situation
     if (