diff --git a/services/native/src/actions/default/display/device_state_action.cpp b/services/native/src/actions/default/display/device_state_action.cpp
index 22156b0fb8a40c9b71969feb1254aabdae9c213d..e4d2eab43c71b0c8e6bc914278b61c9f57b84725 100644
--- a/services/native/src/actions/default/display/device_state_action.cpp
+++ b/services/native/src/actions/default/display/device_state_action.cpp
@@ -237,6 +237,7 @@ uint32_t DeviceStateAction::SetDisplayState(DisplayState state, StateChangeReaso
                 std::string identity = IPCSkeleton::ResetCallingIdentity();
                 // SuspendBegin is processed inside IsInterruptingScreenOff
                 if (IsInterruptingScreenOff(dispReason)) {
+                    IPCSkeleton::SetCallingIdentity(identity);
                     return ActionResult::FAILED;
                 }
                 IPCSkeleton::SetCallingIdentity(identity);