diff --git a/omniadvisor/compile.py b/omniadvisor/compile.py
index ad63aee52bb9b08a85f8571bab948973efed8398..8032bec14e611a786da3e112f5ff8fa453bd21de 100644
--- a/omniadvisor/compile.py
+++ b/omniadvisor/compile.py
@@ -91,10 +91,11 @@ def set_permissions(root_dir: dir):
                 file_path = os.path.join(root, file)
                 os.chmod(file_path, 0o640)
         # 非配置目录，则为程序目录
-        os.chmod(root, 0o550)
-        for file in files:
-            file_path = os.path.join(root, file)
-            os.chmod(file_path, 0o550)
+        else:
+            os.chmod(root, 0o550)
+            for file in files:
+                file_path = os.path.join(root, file)
+                os.chmod(file_path, 0o550)
 
 
 
diff --git a/omniadvisor/script/auto_deploy.sh b/omniadvisor/script/auto_deploy.sh
index 50d8e869267573e0e2a2d3031bc30e240b367599..f0bb5a28a3853477b8e4abe904de8bb2eb696189 100644
--- a/omniadvisor/script/auto_deploy.sh
+++ b/omniadvisor/script/auto_deploy.sh
@@ -49,6 +49,8 @@ cp -r $core_dir/src/algo $frame_dir/src/
 log INFO "执行命令:cp -r $core_dir/config/* $frame_dir/config/ 合并配置文件"
 cp -r $core_dir/config/* $frame_dir/config/
 
+log INFO "执行命令:mkdir -p $frame_dir/kmc_env && cp -r $core_dir/kmc_env/ $frame_dir/ 合并kmc模块"
+mkdir -p $frame_dir/kmc_env && cp -r $core_dir/kmc_env/ $frame_dir/
 
 # 修改spark_submit_modified的文件权限 用readlink获取真实路径防止提权
 spark_submit_modified=$frame_dir/script/spark-submit