diff --git a/pkgs/by-name/im/immich-machine-learning/package.nix b/pkgs/by-name/im/immich-machine-learning/package.nix
index 5b35ecfaea04..1a7255dd91ff 100644
--- a/pkgs/by-name/im/immich-machine-learning/package.nix
+++ b/pkgs/by-name/im/immich-machine-learning/package.nix
@@ -3,6 +3,7 @@
   fetchFromGitHub,
   immich,
   python3,
+  nixosTests,
 }:
 let
   python = python3.override {
@@ -27,7 +28,7 @@ let
     };
   };
 in
-python.pkgs.buildPythonApplication {
+python.pkgs.buildPythonApplication rec {
   pname = "immich-machine-learning";
   inherit (immich) version;
   src = "${immich.src}/machine-learning";
@@ -83,7 +84,7 @@ python.pkgs.buildPythonApplication {
     cp -r ann $out/${python.sitePackages}/
 
     makeWrapper ${lib.getExe python.pkgs.gunicorn} "''${!outputBin}"/bin/machine-learning \
-      --prefix PYTHONPATH : "$out/${python.sitePackages}:$PYTHONPATH" \
+      --prefix PYTHONPATH : "$out/${python.sitePackages}:${python.pkgs.makePythonPath dependencies}" \
       --set-default MACHINE_LEARNING_WORKERS 1 \
       --set-default MACHINE_LEARNING_WORKER_TIMEOUT 120 \
       --set-default MACHINE_LEARNING_CACHE_FOLDER /var/cache/immich \
@@ -96,6 +97,10 @@ python.pkgs.buildPythonApplication {
         --log-config-json $out/share/immich/log_conf.json"
   '';
 
+  passthru.tests = {
+    inherit (nixosTests) immich;
+  };
+
   meta = {
     description = "Self-hosted photo and video backup solution (machine learning component)";
     homepage = "https://immich.app/";