diff --git a/dpgen/__init__.py b/dpgen/__init__.py
index 3fad112f2..6e6e10229 100644
--- a/dpgen/__init__.py
+++ b/dpgen/__init__.py
@@ -11,6 +11,7 @@
 # dlogf_formatter=logging.Formatter('%(asctime)s - %(name)s - [%(filename)s:%(funcName)s - %(lineno)d ] - %(levelname)s \n %(message)s')
 dlogf.setFormatter(dlogf_formatter)
 dlog.addHandler(dlogf)
+logging.basicConfig(level=logging.WARNING)
 
 __author__ = "Han Wang"
 __copyright__ = "Copyright 2019"
diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py
index 81a105b64..fa70d1cc3 100644
--- a/dpgen/generator/run.py
+++ b/dpgen/generator/run.py
@@ -4693,7 +4693,7 @@ def post_fp(iter_index, jdata):
 
 
 def set_version(mdata):
-    deepmd_version = "1"
+    deepmd_version = "2"
     mdata["deepmd_version"] = deepmd_version
     return mdata
 
diff --git a/tests/generator/machine-local-v1.json b/tests/generator/machine-local-v1.json
index 6ffac93a7..0ec4ebe34 100644
--- a/tests/generator/machine-local-v1.json
+++ b/tests/generator/machine-local-v1.json
@@ -1,4 +1,5 @@
 {
+    "deepmd_version": "1",
     "train_machine": {
         "machine_type": "shell",
         "lazy_local": true