diff --git a/flake.nix b/flake.nix
index 7a66712..75cefae 100644
--- a/flake.nix
+++ b/flake.nix
@@ -21,14 +21,16 @@
       ...
     }@inputs:
     flake-parts.lib.mkFlake { inherit inputs; } (
-      { config, pkgs, ... }: {
-        systems = [ "x86_64-linux" ];
+      let
+        theme = "catppuccin-mocha";
+      in
+      {
         imports = [
           wrappers.flakeModules.wrappers
           ./modules/theme.nix
-          ./hosts.nix
+          (import ./hosts.nix { themeName = theme; })
         ];
-    
+
       }
     );
 }
diff --git a/hosts.nix b/hosts.nix
index 02474eb..e47bc07 100644
--- a/hosts.nix
+++ b/hosts.nix
@@ -1,7 +1,4 @@
-{ inputs, config, ... }:
-let
-  themeName = "catppuccin-mocha";
-in
+{themeName}: { inputs,  config, self, ... }:
 {
   flake.nixosConfigurations.nerv = inputs.nixpkgs.lib.nixosSystem {
     specialArgs = { inherit inputs; };
diff --git a/modules/hosts/nerv/configuration.nix b/modules/hosts/nerv/configuration.nix
index d4df98f..e7c84ea 100644
--- a/modules/hosts/nerv/configuration.nix
+++ b/modules/hosts/nerv/configuration.nix
@@ -19,6 +19,9 @@
   networking.hostName = "nerv"; # Define your hostname.
   networking.networkmanager.enable = true;
 
+  # enable flakes
+    nix.settings.experimental-features = [ "nix-command" "flakes" ];
+
   # use proprietary nvidia drivers
   hardware.graphics.enable = true;
   services.xserver.videoDrivers = [ "nvidia" ];
diff --git a/modules/theme.nix b/modules/theme.nix
deleted file mode 100644
index eaf7dca..0000000
--- a/modules/theme.nix
+++ /dev/null
@@ -1,12 +0,0 @@
-{ inputs, lib, pkgs, config, ... }: {
-  options.theme.name = lib.mkOption {
-    type = lib.types.str;
-    default = "${inputs.tt-schemes}/base16/oxocarbon-dark.yaml";
-    example = "${inputs.tt-schemes}/base16/catppuccin-mocha.yaml";
-    description = "set the used theme";
-  };
-  options.theme.mkSchemeAttr = lib.mkOption {
-    type = lib.types.attrSet;
-  };
-  config.theme.name = "${inputs.tt-schemes}/base16/oxocarbon-dark.yaml";
-}