diff --git a/flakeModule.nix b/flakeModule.nix
index 8f2330e..f359ec7 100644
--- a/flakeModule.nix
+++ b/flakeModule.nix
@@ -74,7 +74,7 @@
           id_tokens = mkNullOption (types.attrs);
           "inherit" = mkNullOption (types.attrs);
           interruptible = mkNullOption (types.bool);
-          needs = mkNullOption (types.listOf types.str);
+          needs = mkNullOption (types.listOf (types.either types.str types.attrs));
           publish = mkNullOption (types.str);
           pages = mkNullOption (types.attrs);
           parallel = mkNullOption (types.either types.int types.attrs);