diff --git a/requirements/tests.txt b/requirements/tests.txt
index d74cc6016a..581ae5c080 100644
--- a/requirements/tests.txt
+++ b/requirements/tests.txt
@@ -6,5 +6,6 @@ isort==4.3.21
 openpyxl==3.0.9
 pandas
 pytest
+pyyaml
 xlrd==1.2.0
 yapf
diff --git a/tools/package_tools/mmdeploy_builder.py b/tools/package_tools/mmdeploy_builder.py
index 3a6ee7a912..e1523d021d 100644
--- a/tools/package_tools/mmdeploy_builder.py
+++ b/tools/package_tools/mmdeploy_builder.py
@@ -163,8 +163,8 @@ def build_mmdeploy(cfg, mmdeploy_dir, dist_dir=None):
         build_cmd = 'cmake --build . --config Release -- /m'
         _call_command(build_cmd, build_dir)
         install_cmd = 'cmake --install . --config Release'
-        _remove_if_exist(build_dir, 'lib', 'Release')
         _call_command(install_cmd, build_dir)
+        _remove_if_exist(osp.join(build_dir, 'lib', 'Release'))
     else:
         # build cmd
         build_cmd = 'cmake --build . -- -j$(nproc) && cmake --install .'