From 788a42b341f9a80c2fc0c3f51b6c01e6b3921170 Mon Sep 17 00:00:00 2001
From: Syed Sajjad  Hussain Shah <ssajjad@2u.com>
Date: Mon, 22 Apr 2024 15:17:08 +0500
Subject: [PATCH] fix: fix register button loader for control

---
 src/register/data/optimizelyExperiment/helper.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/register/data/optimizelyExperiment/helper.js b/src/register/data/optimizelyExperiment/helper.js
index 7140d5ec..f3d21381 100644
--- a/src/register/data/optimizelyExperiment/helper.js
+++ b/src/register/data/optimizelyExperiment/helper.js
@@ -85,7 +85,8 @@ export const getRegisterButtonClassInExperiment = (expVariation, registerPageSte
 export const getRegisterButtonSubmitStateInExperiment = (
   registerSubmitState, validationsSubmitState, expVariation, registerPageStep,
 ) => {
-  if (expVariation === MULTI_STEP_REGISTRATION_EXP_VARIATION && registerPageStep !== THIRD_STEP) {
+  if ((expVariation === MULTI_STEP_REGISTRATION_EXP_VARIATION && registerPageStep !== THIRD_STEP)
+    || (expVariation === CONTROL && registerPageStep !== SECOND_STEP)) {
     return validationsSubmitState;
   }
   return registerSubmitState;