diff --git a/src/main.py b/src/main.py
index 135c3af..b80fb21 100644
--- a/src/main.py
+++ b/src/main.py
@@ -95,7 +95,7 @@ def generate_json(output_path,project_id):
         db_url_from = os.getenv("DATABASE_URL")
         engine = create_engine(db_url_from)
         
-        dataset_base = "./src/dataset/agent-v1-c4"
+        dataset_base = "./dataset/agent-v1-c4"
         projects = load_dataset(dataset_base)
 
         project_id = 'shanxuan1212'