diff --git a/api/src/opentrons/protocol_engine/actions/get_state_update.py b/api/src/opentrons/protocol_engine/actions/get_state_update.py
index e0ddadc3222..3805ae642a5 100644
--- a/api/src/opentrons/protocol_engine/actions/get_state_update.py
+++ b/api/src/opentrons/protocol_engine/actions/get_state_update.py
@@ -1,9 +1,12 @@
 # noqa: D100
-
+from __future__ import annotations
+from typing import TYPE_CHECKING
 
 from .actions import Action, SucceedCommandAction, FailCommandAction
 from ..commands.command import DefinedErrorData
-from ..state.update_types import StateUpdate
+
+if TYPE_CHECKING:
+    from ..state.update_types import StateUpdate
 
 
 def get_state_update(action: Action) -> StateUpdate | None: