diff --git a/flake.nix b/flake.nix
index 7209c25..1972027 100644
--- a/flake.nix
+++ b/flake.nix
@@ -97,6 +97,9 @@
         specialArgs = { inherit inputs; };
         modules = [
           ./config/hosts/lain/iso.nix
+          ({ ... }: {
+            nixpkgs.overlays = [ self.overlays.kyouma self.overlays.otherPkgs ];
+          })
         ];
       };
     };
diff --git a/pkgs/upgrade-system/upgrade-system.sh b/pkgs/upgrade-system/upgrade-system.sh
index 8c721a6..de09135 100644
--- a/pkgs/upgrade-system/upgrade-system.sh
+++ b/pkgs/upgrade-system/upgrade-system.sh
@@ -13,7 +13,8 @@ while [[ $# -gt 0 ]]; do
       shift
     ;;
     --branch)
-      BRANCH=$2
+      BRANCH="$2"
+      shift
       shift
     ;;
     --no-delay)