diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index cd7d1f3cdff3..2439aff16205 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -85,6 +85,8 @@ jobs:
           sudo pip2 install sonic_platform_common-1.0-py2-none-any.whl
           sudo pip3 install swsssdk-2.0.1-py3-none-any.whl
           sudo pip3 install sonic_py_common-1.0-py3-none-any.whl
+          sudo pip3 install sonic_yang_mgmt-1.0-py3-none-any.whl
+          sudo pip3 install sonic_yang_models-1.0-py3-none-any.whl
           sudo pip3 install sonic_config_engine-1.0-py3-none-any.whl
           sudo pip3 install sonic_platform_common-1.0-py3-none-any.whl
         workingDirectory: $(Pipeline.Workspace)/target/python-wheels/