diff --git a/flake.nix b/flake.nix
index cb0fcdd..1acdd75 100644
--- a/flake.nix
+++ b/flake.nix
@@ -18,17 +18,20 @@
   outputs =
     inputs@{ self, nixpkgs, ... }:
     let
-      hosts = [
-        "work-ntb"
-        "lenar"
-      ];
+      hosts = {
+        "work-ntb" = "x86_64-linux";
+        "lenar" = "x86_64-linux";
+      };
     in
     {
       # Please replace my-nixos with your hostname
-      nixosConfigurations = nixpkgs.lib.genAttrs hosts (
+      nixosConfigurations = nixpkgs.lib.genAttrs (builtins.attrNames hosts) (
         host:
+        let
+          arch = hosts.${host};
+        in
         nixpkgs.lib.nixosSystem {
-          system = "x86_64-linux";
+          system = arch;
           modules = [
             ./hosts/${host}
             ./hosts/base