diff --git a/src/MaaWpfGui/Main/SettingsViewModel.cs b/src/MaaWpfGui/Main/SettingsViewModel.cs
index 1f66c85e5a..7d7557f9f5 100644
--- a/src/MaaWpfGui/Main/SettingsViewModel.cs
+++ b/src/MaaWpfGui/Main/SettingsViewModel.cs
@@ -2284,9 +2284,11 @@ namespace MaaWpfGui
             set
             {
                 SetAndNotify(ref _customStageCode, value);
-                ViewStatusStorage.Set("GUI.CustomStageCode", value.ToString());
                 var mainModel = _container.Get<TaskQueueViewModel>();
+                mainModel.RemainingSanityStageDisplay1 = UseRemainingSanityStage && !value;
+                mainModel.RemainingSanityStageDisplay2 = UseRemainingSanityStage && value;
                 mainModel.CustomStageCode = value;
+                ViewStatusStorage.Set("GUI.CustomStageCode", value.ToString());
             }
         }