diff --git a/README.md b/README.md
index 403c70d..e88e8d0 100644
--- a/README.md
+++ b/README.md
@@ -76,7 +76,6 @@ nhmu
           enableShell = true;
           _1password = true;
           docker = true;
-          # graphical = true;
           graphical = {
             steam = true;
             vscode = true;
diff --git a/flake.nix b/flake.nix
index b05b545..af8dabf 100644
--- a/flake.nix
+++ b/flake.nix
@@ -22,7 +22,10 @@
 
       # These will get applied to both the configured user and the root user (if enabled)
       userOptions = config: {
-        openssh.authorizedKeys.keyFiles = lib.optionals config.ssh [ ./personal_keys ];
+        openssh.authorizedKeys = lib.mkIf config.ssh (lib.mkMerge [
+          (lib.mkIf (config.profile == "personal") { keyFiles = [ ./personal ]; })
+          (lib.mkIf (config.profile == "work") { keyFiles = [ ./work ]; })
+        ]);
         shell = lib.mkIf config.enableShell pkgs.zsh;
       };
 
diff --git a/personal_keys b/keys/personal
similarity index 100%
rename from personal_keys
rename to keys/personal
diff --git a/work_keys b/keys/work
similarity index 100%
rename from work_keys
rename to keys/work