diff --git a/nixos/modules/zsh.nix b/nixos/modules/zsh.nix
index 63ef5dc..da04a6e 100644
--- a/nixos/modules/zsh.nix
+++ b/nixos/modules/zsh.nix
@@ -1,4 +1,4 @@
-{ pkgs, ... }: {
+{ pkgs, vars, ... }: {
 
   environment.systemPackages = with pkgs; [
     zsh
@@ -6,7 +6,7 @@
   ];
 
   environment.variables = {
-    NIX_FLAKE_PATH = "/etc/nixos";
+    NIX_FLAKE_PATH = "$HOME/.nix/";
   };
 
   programs.zsh = {
@@ -14,8 +14,6 @@
     enableBashCompletion = true;
     enableCompletion = true;
     autosuggestions.enable = true;
-    # history.append = true;
-    # history.size = 10000;
 
     ohMyZsh = {
       enable = true;
@@ -28,7 +26,8 @@
 
     shellAliases = {
       ll = "ls -la";
-      update = "sudo nixos-rebuild switch --upgrade-all --flake $NIX_FLAKE_PATH";
+      update = "sudo nixos-rebuild switch --upgrade-all --flake $NIX_FLAKE_PATH#${vars.hostName}";
+      rebuild = "sudo nixos-rebuild switch --flake $NIX_FLAKE_PATH#${vars.hostName}";
     };
   };