Add check if current step doesn't exist and no session current step t…

…hen get the current step
silverstripe · Nov 28, 2019 · 2574126 · 2574126
1 parent a57bfe7
commit 2574126
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/Forms/MultiForm.php b/src/Forms/MultiForm.php
@@ -229,14 +229,14 @@ public function getCurrentStep()
         $currentStep = null;
         $StepID = $this->controller->getRequest()->getVar('StepID');
         if (isset($StepID)) {
-            $currentStep = DataObject::get_one(
-                MultiFormStep::class,
-                [
-                    'SessionID' => $this->session->ID,
-                    'ID' => $StepID
-                ]
-            );
-        } elseif ($this->session->CurrentStepID) {
+            $currentStep = MultiFormStep::get()->filter([
+                'SessionID' => $this->session->ID,
+                'ID' => $StepID
+            ])->first();
+        }
+
+        // if current step doesn't exist and no session current step then get the current step
+        if (!$currentStep && $this->session->CurrentStepID) {
             $currentStep = $this->session->CurrentStep();
         }