diff --git a/config/home-manager/flake.nix b/config/home-manager/flake.nix
index cb53c15..1601953 100644
--- a/config/home-manager/flake.nix
+++ b/config/home-manager/flake.nix
@@ -4,7 +4,7 @@
   inputs = {
     nixpkgs.url = "github:nixos/nixpkgs/nixos-24.05";
 
-    # Pinned versions for specific packages
+    # Pinned versions for specific packages (https://nixhub.io)
     nixpkgs-go.url = "github:nixos/nixpkgs/d4f247e89f6e10120f911e2e2d2254a050d0f732";
     nixpkgs-vscode.url = "github:nixos/nixpkgs/d4f247e89f6e10120f911e2e2d2254a050d0f732";
     nixpkgs-zed.url = "github:nixos/nixpkgs/41dea55321e5a999b17033296ac05fe8a8b5a257";
diff --git a/setup.sh b/setup.sh
index 4002437..98122d9 100755
--- a/setup.sh
+++ b/setup.sh
@@ -155,7 +155,7 @@ install_nix
 install_home_manager
 
 # Rebuild NixOS
-cd $HOME/dotfiles/config/nixos && sudo nixos-rebuild switch --flake .#$DOTF_HOSTNAME --impure
+cd $HOME/dotfiles/config/nixos && sudo nixos-rebuild switch --flake .#$hostname --impure
 if [ $? -ne 0 ]; then
     tput setaf 1
     echo "Failed to rebuild NixOS. Exiting..."
@@ -164,7 +164,7 @@ if [ $? -ne 0 ]; then
 fi
 
 # Rebuild Home Manager
-cd $HOME/dotfiles/config/home-manager && NIXPKGS_ALLOW_UNFREE=1 home-manager switch --flake .#$DOTF_HOSTNAME --impure
+cd $HOME/dotfiles/config/home-manager && NIXPKGS_ALLOW_UNFREE=1 home-manager switch --flake .#$hostname --impure
 if [ $? -ne 0 ]; then
     tput setaf 1
     echo "Failed to rebuild Home Manager. Exiting..."