diff --git a/iwf/state_decision.py b/iwf/state_decision.py
index dad89ec..439858e 100644
--- a/iwf/state_decision.py
+++ b/iwf/state_decision.py
@@ -2,7 +2,7 @@
 
 import typing
 
-from iwf_api.models import WorkflowConditionalClose, WorkflowConditionalCloseType
+from iwf.iwf_api.models import WorkflowConditionalClose, WorkflowConditionalCloseType
 
 if typing.TYPE_CHECKING:
     from iwf.registry import Registry