diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile
index d8e014632ccf8..7dc206ab55e96 100644
--- a/machine-learning/Dockerfile
+++ b/machine-learning/Dockerfile
@@ -115,7 +115,7 @@ ARG BUILD_SOURCE_COMMIT
 ENV IMMICH_BUILD=${BUILD_ID}
 ENV IMMICH_BUILD_URL=https://github.com/immich-app/immich/actions/runs/${BUILD_ID}
 ENV IMMICH_BUILD_IMAGE=${BUILD_IMAGE}
-ENV IMMICH_BUILD_IMAGE_URL=https://github.com/immich-app/immich/pkgs/container/immich-server
+ENV IMMICH_BUILD_IMAGE_URL=https://github.com/immich-app/immich/pkgs/container/immich-machine-learning
 ENV IMMICH_REPOSITORY=immich-app/immich
 ENV IMMICH_REPOSITORY_URL=https://github.com/immich-app/immich
 ENV IMMICH_SOURCE_REF=${BUILD_SOURCE_REF}