diff --git a/keys/flake-part.nix b/keys/flake-part.nix
index ddc1f929..cad4ee27 100644
--- a/keys/flake-part.nix
+++ b/keys/flake-part.nix
@@ -1 +1,5 @@
-{ _module.args.keys = import ./.; }
+{
+  _class = "flake";
+
+  _module.args.keys = import ./.;
+}
diff --git a/secrets/flake-part.nix b/secrets/flake-part.nix
index 4201f041..0d41cdbd 100644
--- a/secrets/flake-part.nix
+++ b/secrets/flake-part.nix
@@ -1 +1,5 @@
-{ _module.args.secrets = import ./.; }
+{
+  _class = "flake";
+
+  _module.args.secrets = import ./.;
+}