diff --git a/deploy.sh b/deploy.sh
index b8ac4e8..c82463f 100755
--- a/deploy.sh
+++ b/deploy.sh
@@ -252,6 +252,7 @@ files_generation() {
   cp -ravu ./nix-config/* extra-files/etc/nixos/
 }
 
+disk_config_generation() {
 ### --> Disk wiping warning
   gum style --border normal --margin "1" --padding "1 2" --border-foreground 212 "
   ⚠️  $(gum style --foreground 212 'WARNING:') You will choose the disks to install NixOS on.
@@ -265,7 +266,7 @@ files_generation() {
 ### Disk wiping warning <--
 
 ### --> Get disk information
-  DISK_DETAILS=$(ssh nixos@192.168.11.115 "
+  DISK_DETAILS=$(ssh_to_host "
 # Declare arrays and variables
 HDD=1