3 files changed, 4 insertions, 4 deletions
diff --git a/lib/tests/modules/class-check.nix b/lib/tests/modules/class-check.nix
index 7874d0e28ec73..293fd4abd6286 100644
--- a/lib/tests/modules/class-check.nix
+++ b/lib/tests/modules/class-check.nix
@@ -63,14 +63,14 @@
         modules = [
           ./module-class-is-nixos.nix
           { _file = "foo.nix#darwinModules.default";
-            class = "darwin";
+            _class = "darwin";
             config = {};
             imports = [];
           }
         ];
       };
 
-    sub.nixosOk = { config = {}; class = "nixos"; };
+    sub.nixosOk = { _class = "nixos"; };
     sub.nixosFail = { imports = [ ./module-class-is-darwin.nix ]; };
   };
 }
diff --git a/lib/tests/modules/module-class-is-darwin.nix b/lib/tests/modules/module-class-is-darwin.nix
index d8b60203f707e..bacf45626d714 100644
--- a/lib/tests/modules/module-class-is-darwin.nix
+++ b/lib/tests/modules/module-class-is-darwin.nix
@@ -1,4 +1,4 @@
 {
-  class = "darwin";
+  _class = "darwin";
   config = {};
 }
diff --git a/lib/tests/modules/module-class-is-nixos.nix b/lib/tests/modules/module-class-is-nixos.nix
index 04b6e860e8900..6d62feedae486 100644
--- a/lib/tests/modules/module-class-is-nixos.nix
+++ b/lib/tests/modules/module-class-is-nixos.nix
@@ -1,4 +1,4 @@
 {
-  class = "nixos";
+  _class = "nixos";
   config = {};
 }