diff --git a/api/core/apps/config_validators/agent.py b/api/core/apps/config_validators/agent.py
index c6584d2903..b445aedbf8 100644
--- a/api/core/apps/config_validators/agent.py
+++ b/api/core/apps/config_validators/agent.py
@@ -1,7 +1,7 @@
 import uuid
 
-from core.agent.agent_executor import PlanningStrategy
 from core.apps.config_validators.dataset import DatasetValidator
+from core.entities.agent_entities import PlanningStrategy
 
 OLD_TOOLS = ["dataset", "google_search", "web_reader", "wikipedia", "current_datetime"]
 
diff --git a/api/core/apps/config_validators/dataset.py b/api/core/apps/config_validators/dataset.py
index 9846f9085c..fb5b648320 100644
--- a/api/core/apps/config_validators/dataset.py
+++ b/api/core/apps/config_validators/dataset.py
@@ -1,6 +1,6 @@
 import uuid
 
-from core.agent.agent_executor import PlanningStrategy
+from core.entities.agent_entities import PlanningStrategy
 from models.model import AppMode
 from services.dataset_service import DatasetService