diff --git a/.buildkite/scripts/simple_test.sh b/.buildkite/scripts/simple_test.sh
index 6c406d095..58379c964 100755
--- a/.buildkite/scripts/simple_test.sh
+++ b/.buildkite/scripts/simple_test.sh
@@ -49,4 +49,4 @@ VENV_PYTHON="${VENV_DIR}/bin/python"
 
 "${UV_BIN}" pip install --python "${VENV_PYTHON}" vllm==0.11.0
 "${UV_BIN}" pip install --python "${VENV_PYTHON}" -e ".[dev]"
-"${VENV_PYTHON}" -m pytest -v -s tests/test_omni_llm.py
+"${VENV_PYTHON}" -m pytest -v -s tests/entrypoints/
diff --git a/tests/test_omni_llm.py b/tests/entrypoints/test_omni_llm.py
similarity index 100%
rename from tests/test_omni_llm.py
rename to tests/entrypoints/test_omni_llm.py