diff --git a/flake.nix b/flake.nix
index f75cf19..d7e1de5 100644
--- a/flake.nix
+++ b/flake.nix
@@ -17,7 +17,7 @@
       };
 
       overlays = {
-        default = (final: _: { dzgui = (final.callPackage ./package { }); });
+        default = import ./overlay.nix;
       };
 
       nixosModules = rec {
@@ -32,15 +32,7 @@
       checks.${system}.nixosCheck =
         (nixpkgs.lib.nixosSystem {
           inherit system;
-          modules = [
-            self.nixosModules.default
-            {
-              programs.dzgui.enable = true;
-
-              boot.isContainer = true;
-              system.stateVersion = "24.05";
-            }
-          ];
+          modules = [ ./nixos-test/configuration.nix ];
         }).config.system.build.toplevel;
     };
 }
diff --git a/nixos-test/configuration.nix b/nixos-test/configuration.nix
new file mode 100644
index 0000000..302d4d3
--- /dev/null
+++ b/nixos-test/configuration.nix
@@ -0,0 +1,7 @@
+{
+  imports = [ ../module.nix ];
+  programs.dzgui.enable = true;
+
+  boot.isContainer = true;
+  system.stateVersion = "24.05";
+}
diff --git a/overlay.nix b/overlay.nix
new file mode 100644
index 0000000..a431f2a
--- /dev/null
+++ b/overlay.nix
@@ -0,0 +1 @@
+(final: _: { dzgui = (final.callPackage ./package { }); })