diff --git a/window_scene/test/unittest/multi_user/session_manager_lite_test.cpp b/window_scene/test/unittest/multi_user/session_manager_lite_test.cpp
index a7e910c7a7591a9e388b0edb41ba2e031e2017f6..07a29f1ed5965e72bd9cede9565877769d8971da 100644
--- a/window_scene/test/unittest/multi_user/session_manager_lite_test.cpp
+++ b/window_scene/test/unittest/multi_user/session_manager_lite_test.cpp
@@ -111,11 +111,10 @@ HWTEST_F(SessionManagerLiteTest, OnWMSConnectionChangedCallback, Function | Smal
 {
     ASSERT_NE(nullptr, sml_);
     bool funcInvoked = false;
-    sml_->wmsConnectionChangedFunc_ = nullptr;
-    sml_->OnWMSConnectionChangedCallback(101, DEFAULT_SCREEN_ID, true, false);
-
     sml_->wmsConnectionChangedFunc_ = [&](int32_t userId, int32_t screenId, bool isConnected) { funcInvoked = true; };
     sml_->OnWMSConnectionChangedCallback(101, DEFAULT_SCREEN_ID, true, true);
+    sml_->OnWMSConnectionChangedCallback(101, DEFAULT_SCREEN_ID, true, false);
+    sml_->wmsConnectionChangedFunc_ = nullptr;
     ASSERT_EQ(funcInvoked, true);
 }
 
@@ -177,6 +176,7 @@ HWTEST_F(SessionManagerLiteTest, OnUserSwitch, Function | SmallTest | Level2)
     auto sessionManagerService = sml_->GetSessionManagerServiceProxy();
     sml_->OnUserSwitch(sessionManagerService);
     ASSERT_EQ(funInvoked, true);
+    sml_->userSwitchCallbackFunc_ = nullptr;
 }
 
 /**
diff --git a/window_scene/test/unittest/multi_user/session_manager_test.cpp b/window_scene/test/unittest/multi_user/session_manager_test.cpp
index 73cd460b161832620d78f5d717ba1e2c81f726f6..2e7a8ef63b59a654f9da816ce57020d5350f5959 100644
--- a/window_scene/test/unittest/multi_user/session_manager_test.cpp
+++ b/window_scene/test/unittest/multi_user/session_manager_test.cpp
@@ -82,12 +82,12 @@ HWTEST_F(SessionManagerTest, ClearSessionManagerProxy, Function | SmallTest | Le
 HWTEST_F(SessionManagerTest, OnWMSConnectionChangedCallback, Function | SmallTest | Level2)
 {
     auto sm = SessionManager::GetInstance(-1);
+    ASSERT_NE(nullptr, sm);
     bool funcInvoked = false;
-    sm->wmsConnectionChangedFunc_ = nullptr;
-    sm->OnWMSConnectionChangedCallback(101, DEFAULT_SCREEN_ID, true, false);
-
     sm->wmsConnectionChangedFunc_ = [&](int32_t userId, int32_t screenId, bool isConnected) { funcInvoked = true; };
     sm->OnWMSConnectionChangedCallback(101, DEFAULT_SCREEN_ID, true, true);
+    sm->OnWMSConnectionChangedCallback(101, DEFAULT_SCREEN_ID, true, false);
+    sm->wmsConnectionChangedFunc_ = nullptr;
     ASSERT_EQ(funcInvoked, true);
 }
 
@@ -163,6 +163,7 @@ HWTEST_F(SessionManagerTest, OnUserSwitch, Function | SmallTest | Level2)
     auto sessionManagerService = SessionManagerLite::GetInstance().GetSessionManagerServiceProxy();
     sm->OnUserSwitch(sessionManagerService);
     ASSERT_EQ(funInvoked, true);
+    sml_->userSwitchCallbackFunc_ = nullptr;
 }
 
 /**