diff --git a/scripts/install b/scripts/install
index 9c40147..0371e10 100755
--- a/scripts/install
+++ b/scripts/install
@@ -160,7 +160,7 @@ ynh_secure_remove --file="$final_path/build/output"
 # Install immich machine learning
 pushd $final_path/build
         detect_arch
-	./docker-image-extract altran1502/immich-web:$IMMICH_ML_VERSION
+	./docker-image-extract altran1502/immich-machine-learning:$IMMICH_ML_VERSION
 popd
 mkdir -p "$final_path/ml/dist/"
 mkdir -p "$final_path/ml/node_modules/"