diff --git a/omagent-core/src/omagent_core/engine/http/models/task.py b/omagent-core/src/omagent_core/engine/http/models/task.py
index cceb0813..3c0a2523 100644
--- a/omagent-core/src/omagent_core/engine/http/models/task.py
+++ b/omagent-core/src/omagent_core/engine/http/models/task.py
@@ -2,7 +2,7 @@
 import re  # noqa: F401
 
 import six
-from omagent_core.engine.http.models import WorkflowTask
+from omagent_core.engine.http.models.workflow_task import WorkflowTask
 from omagent_core.engine.http.models.task_result import TaskResult
 from omagent_core.engine.http.models.task_result_status import TaskResultStatus
 
diff --git a/omagent-core/src/omagent_core/engine/http/models/workflow_def.py b/omagent-core/src/omagent_core/engine/http/models/workflow_def.py
index 801206dd..30ba1bdf 100644
--- a/omagent-core/src/omagent_core/engine/http/models/workflow_def.py
+++ b/omagent-core/src/omagent_core/engine/http/models/workflow_def.py
@@ -5,7 +5,7 @@
 
 import six
 from omagent_core.engine.helpers.helper import ObjectMapper
-from omagent_core.engine.http.models import WorkflowTask
+from omagent_core.engine.http.models.workflow_task import WorkflowTask
 
 object_mapper = ObjectMapper()