diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml
index 3517262a117e..4edd1cadfb2f 100644
--- a/.buildkite/test-pipeline.yaml
+++ b/.buildkite/test-pipeline.yaml
@@ -81,8 +81,9 @@ steps:
 - label: Inputs Test
   #mirror_hardwares: [amd]
   commands:
-  - pytest -v -s test_inputs.py
-  - pytest -v -s multimodal
+    - bash ../.buildkite/download-images.sh
+    - pytest -v -s test_inputs.py
+    - pytest -v -s multimodal
 
 - label: Kernels Test %N
   #mirror_hardwares: [amd]