diff --git a/.github/workflows/buildAndTestRyzenAISw.yml b/.github/workflows/buildAndTestRyzenAISw.yml
index db8d03641a..28e685e303 100644
--- a/.github/workflows/buildAndTestRyzenAISw.yml
+++ b/.github/workflows/buildAndTestRyzenAISw.yml
@@ -97,6 +97,7 @@ jobs:
             pip install onnxruntime==1.17.0
             pip install -r python/requirements.txt
             pip install -r python/requirements_ml.txt
+            HOST_MLIR_PYTHON_PACKAGE_PREFIX=aie pip install -r python/requirements_extras.txt
 
             VERSION=$(utils/clone-llvm.sh --get-wheel-version)