diff --git a/lib.nix b/lib.nix
index b76ce7a..2ff8bb0 100644
--- a/lib.nix
+++ b/lib.nix
@@ -174,7 +174,11 @@ let
       options = {
         flake = flake-parts-lib.mkSubmoduleOptions {
           ${name} = mkOption {
-            type = types.lazyAttrsOf option.type;
+            type = types.attrsWith {
+              elemType = option.type;
+              lazy = true;
+              placeholder = "system";
+            };
             default = { };
             description = ''
               See {option}`perSystem.${name}` for description and examples.