diff --git a/openpype/hosts/maya/api/lib.py b/openpype/hosts/maya/api/lib.py
index 0c571d41e0..7c49c837e9 100644
--- a/openpype/hosts/maya/api/lib.py
+++ b/openpype/hosts/maya/api/lib.py
@@ -146,15 +146,17 @@ def suspended_refresh(suspend=True):
 
     cmds.ogs(pause=True) is a toggle so we cant pass False.
     """
-    original_state = None
-    if not IS_HEADLESS:
-        original_state = cmds.ogs(query=True, pause=True)
+    if IS_HEADLESS:
+        yield
+        return
+
+    original_state = cmds.ogs(query=True, pause=True)
     try:
-        if suspend and not original_state and not IS_HEADLESS:
+        if suspend and not original_state:
             cmds.ogs(pause=True)
         yield
     finally:
-        if suspend and not original_state and not IS_HEADLESS:
+        if suspend and not original_state:
             cmds.ogs(pause=True)