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();