From af91d73036ed02731c1aa6cfaf9da00e6e8e2c02 Mon Sep 17 00:00:00 2001
From: uye <2396806385@qq.com>
Date: Fri, 10 Feb 2023 21:57:48 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=89=A9=E4=BD=99?=
 =?UTF-8?q?=E7=90=86=E6=99=BA=E6=98=BE=E7=A4=BA=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/MaaWpfGui/Main/SettingsViewModel.cs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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