diff --git a/tests/unit/test_release_scripts.py b/tests/unit/test_release_scripts.py
index 5bf3c4d6c..142b03682 100644
--- a/tests/unit/test_release_scripts.py
+++ b/tests/unit/test_release_scripts.py
@@ -29,7 +29,7 @@ def test_deploy_production_uses_production_infra_and_project():
     assert "Production deploy must not build frontend assets on the server" in script
     assert "WEB_DOCKERFILE=infra/docker/web-artifact.Dockerfile" in script
     assert "WEB_NGINX_CONF=infra/docker/nginx-production.conf" in script
-    assert "Skipping production API/worker image builds" in script
+    assert "Skipping production API/worker/web image builds" in script
     assert "RELEASE_VERSION" in script
     assert "RUNTIME_IMAGE_TAR" in script
     assert 'docker load -i "$RUNTIME_IMAGE_TAR"' in script
@@ -239,7 +239,8 @@ def test_gitea_production_deploy_requires_runtime_builder_job():
     assert "runtime-images-${RELEASE_VERSION}.tar" in production_section
     assert "apps/web/dist/index.html" in production_section
     assert (
-        "HOST_PREFIX= sh /var/lib/xiaoxia-saas-production/repo/infra/docker/deploy-production.sh" in production_section
+        'HOST_PREFIX= WEB_IMAGE="xiaoxia-saas-web:${RELEASE_VERSION}" WEB_IMAGE_TAR="/var/lib/xiaoxia-saas-production/web-${RELEASE_VERSION}.tar" sh /var/lib/xiaoxia-saas-production/repo/infra/docker/deploy-production.sh'
+        in production_section
     )