diff --git a/examples/aflow/README.md b/examples/aflow/README.md
index 4842b2044..c2abdff48 100644
--- a/examples/aflow/README.md
+++ b/examples/aflow/README.md
@@ -54,7 +54,7 @@ ## Quick Start
 4. When you first run, you can download the datasets and initial rounds by setting `download(["datasets", "initial_rounds"])` in `examples/aflow/optimize.py`
 5. (Optional) Add your custom dataset and corresponding evaluation function following the [Custom Datasets](#custom-datasets) section
 6. (Optional) If you want to use a portion of the validation data, you can set `va_list` in `examples/aflow/evaluator.py`
-6. Run `python examples/aflow/optimize.py` to start the optimization process!
+6. Run `python -m examples.aflow.optimize` to start the optimization process!
 
 
 ## Reproduce the Results in the Paper
diff --git a/examples/aflow/experiments/optimize_drop.py b/examples/aflow/experiments/optimize_drop.py
index 1b3bcdbcf..80342e1a6 100644
--- a/examples/aflow/experiments/optimize_drop.py
+++ b/examples/aflow/experiments/optimize_drop.py
@@ -3,19 +3,6 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
-import os
-import sys
-
-
-def setup_environment():
-    current_path = os.path.abspath(__file__)
-    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
-    sys.path.insert(0, root_path)
-    os.chdir(root_path)
-
-
-setup_environment()
-
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
 
diff --git a/examples/aflow/experiments/optimize_gsm8k.py b/examples/aflow/experiments/optimize_gsm8k.py
index aefc5ef73..3895838ff 100644
--- a/examples/aflow/experiments/optimize_gsm8k.py
+++ b/examples/aflow/experiments/optimize_gsm8k.py
@@ -3,19 +3,6 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
-import os
-import sys
-
-
-def setup_environment():
-    current_path = os.path.abspath(__file__)
-    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
-    sys.path.insert(0, root_path)
-    os.chdir(root_path)
-
-
-setup_environment()
-
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
 
diff --git a/examples/aflow/experiments/optimize_hotpotqa.py b/examples/aflow/experiments/optimize_hotpotqa.py
index 4a592d116..447759f07 100644
--- a/examples/aflow/experiments/optimize_hotpotqa.py
+++ b/examples/aflow/experiments/optimize_hotpotqa.py
@@ -3,19 +3,6 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
-import os
-import sys
-
-
-def setup_environment():
-    current_path = os.path.abspath(__file__)
-    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
-    sys.path.insert(0, root_path)
-    os.chdir(root_path)
-
-
-setup_environment()
-
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
 
diff --git a/examples/aflow/experiments/optimize_humaneval.py b/examples/aflow/experiments/optimize_humaneval.py
index 965f7b355..90a051aab 100644
--- a/examples/aflow/experiments/optimize_humaneval.py
+++ b/examples/aflow/experiments/optimize_humaneval.py
@@ -3,19 +3,6 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
-import os
-import sys
-
-
-def setup_environment():
-    current_path = os.path.abspath(__file__)
-    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
-    sys.path.insert(0, root_path)
-    os.chdir(root_path)
-
-
-setup_environment()
-
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
 
diff --git a/examples/aflow/experiments/optimize_math.py b/examples/aflow/experiments/optimize_math.py
index c95e0ad87..60e54d64a 100644
--- a/examples/aflow/experiments/optimize_math.py
+++ b/examples/aflow/experiments/optimize_math.py
@@ -3,19 +3,6 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
-import os
-import sys
-
-
-def setup_environment():
-    current_path = os.path.abspath(__file__)
-    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
-    sys.path.insert(0, root_path)
-    os.chdir(root_path)
-
-
-setup_environment()
-
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
 
diff --git a/examples/aflow/experiments/optimize_mbpp.py b/examples/aflow/experiments/optimize_mbpp.py
index 5f7668762..b7e7d306f 100644
--- a/examples/aflow/experiments/optimize_mbpp.py
+++ b/examples/aflow/experiments/optimize_mbpp.py
@@ -3,19 +3,6 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
-import os
-import sys
-
-
-def setup_environment():
-    current_path = os.path.abspath(__file__)
-    root_path = os.path.dirname(os.path.dirname(os.path.dirname(current_path)))
-    sys.path.insert(0, root_path)
-    os.chdir(root_path)
-
-
-setup_environment()
-
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
 
diff --git a/examples/aflow/optimize.py b/examples/aflow/optimize.py
index dafa5fffd..d2bfbc775 100644
--- a/examples/aflow/optimize.py
+++ b/examples/aflow/optimize.py
@@ -3,19 +3,6 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
-import os
-import sys
-
-
-def setup_environment():
-    current_path = os.path.abspath(__file__)
-    root_path = os.path.dirname(os.path.dirname(os.path.dirname(current_path)))
-    sys.path.insert(0, root_path)
-    os.chdir(root_path)
-
-
-setup_environment()
-
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.data.download_data import download
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType