diff --git a/default.nix b/default.nix
index 3fe2bbb..4a81f03 100644
--- a/default.nix
+++ b/default.nix
@@ -16,6 +16,6 @@ in {
   types = types;
   create = cfg: types.diskoLib.create (eval cfg).config.devices;
   mount = cfg: types.diskoLib.mount (eval cfg).config.devices;
-  config = cfg: types.diskoLib.config (eval cfg).config.devices;
+  config = cfg: { imports = types.diskoLib.config (eval cfg).config.devices; };
   packages = cfg: types.diskoLib.packages (eval cfg).config.devices;
 }
diff --git a/types.nix b/types.nix
index db70ede..6c35de6 100644
--- a/types.nix
+++ b/types.nix
@@ -156,9 +156,7 @@ rec {
 
        config :: types.devices -> nixosConfig
     */
-    config = devices: {
-      imports = flatten (map (dev: dev._config) (flatten (map attrValues (attrValues devices))));
-    };
+    config = devices: flatten (map (dev: dev._config) (flatten (map attrValues (attrValues devices))));
     /* Takes a disko device specification and returns a function to get the needed packages to format/mount the disks
 
        packages :: types.devices -> pkgs -> [ derivation ]