From f5236426eeae950ca438b7ff84d8f4f70043d5ee Mon Sep 17 00:00:00 2001
From: Anshuman Pandey <54475686+pandeymangg@users.noreply.github.com>
Date: Sat, 27 Apr 2024 02:03:07 +0530
Subject: [PATCH] fix: survey switch (#2547)

---
 .../surveys/[surveyId]/edit/components/SurveyMenuBar.tsx    | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/apps/web/app/(app)/environments/[environmentId]/surveys/[surveyId]/edit/components/SurveyMenuBar.tsx b/apps/web/app/(app)/environments/[environmentId]/surveys/[surveyId]/edit/components/SurveyMenuBar.tsx
index 04118c00d6..ecc1cd420a 100644
--- a/apps/web/app/(app)/environments/[environmentId]/surveys/[surveyId]/edit/components/SurveyMenuBar.tsx
+++ b/apps/web/app/(app)/environments/[environmentId]/surveys/[surveyId]/edit/components/SurveyMenuBar.tsx
@@ -165,9 +165,11 @@ export const SurveyMenuBar = ({
       });
 
       const segment = await handleSegmentUpdate();
-      await updateSurveyAction({ ...localSurvey, segment });
+      const updatedSurvey = await updateSurveyAction({ ...localSurvey, segment });
+
       setIsSurveySaving(false);
-      setLocalSurvey(localSurvey);
+      setLocalSurvey(updatedSurvey);
+
       toast.success("Changes saved.");
       if (shouldNavigateBack) {
         router.back();