diff --git a/immich-ml-container.nix b/immich-ml-container.nix
index a384952..de949a7 100644
--- a/immich-ml-container.nix
+++ b/immich-ml-container.nix
@@ -1,7 +1,7 @@
 { config, lib, pkgs, ... }:
 
+with lib;
 let cfg = config.services.immichMlContainer;
-
 in {
   options.services.immichMlContainer = with types; {
     enable = mkEnableOption "Enable machine learning container.";