diff --git a/README.md b/README.md
index 0edf0c6..ad289dd 100644
--- a/README.md
+++ b/README.md
@@ -3,8 +3,9 @@ tags:
 - llama2
 - Agent
 - AgentLM
-datasets: 
-- ZhipuAI/AgentInstruct
+datasets:
+  train:
+  - ZhipuAI/AgentInstruct
 ---
 
 ## AgentLM-13B