From 1e3fdde6d8cb65b385b0992efa5e3a517faacb86 Mon Sep 17 00:00:00 2001
From: Robert Wloch <99718767+robert-wloch-iits@users.noreply.github.com>
Date: Sat, 11 Feb 2023 16:45:36 +0100
Subject: [PATCH] fix(hooks): [use-lockscreen] prevent flaky unit tests
 (#11404)

---
 packages/hooks/use-lockscreen/index.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/hooks/use-lockscreen/index.ts b/packages/hooks/use-lockscreen/index.ts
index 41999acafd..d83e16f9d2 100644
--- a/packages/hooks/use-lockscreen/index.ts
+++ b/packages/hooks/use-lockscreen/index.ts
@@ -40,8 +40,8 @@ export const useLockscreen = (trigger: Ref<boolean>) => {
 
   const cleanup = () => {
     setTimeout(() => {
-      removeClass(document.body, hiddenCls.value)
-      if (withoutHiddenClass) {
+      removeClass(document?.body, hiddenCls.value)
+      if (withoutHiddenClass && document) {
         document.body.style.width = bodyWidth
       }
     }, 200)