diff --git a/host/003 - install-amd-drivers.sh b/host/003 - install-amd-drivers.sh
index 984bc3b..37752b2 100755
--- a/host/003 - install-amd-drivers.sh	
+++ b/host/003 - install-amd-drivers.sh	
@@ -9,8 +9,8 @@ wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
 gpg --dearmor | tee /etc/apt/keyrings/rocm.gpg > /dev/null
 
 tee /etc/apt/sources.list.d/rocm.list << EOF
-deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/7.1 noble main
-deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/graphics/7.1/ubuntu noble main
+deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/7.1.1 noble main
+deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/graphics/7.1.1/ubuntu noble main
 EOF
 
 tee /etc/apt/preferences.d/rocm-pin-600 << EOF
diff --git a/host/031 - create-gpu-lxc.sh b/host/031 - create-gpu-lxc.sh
index 165c597..1f292a5 100755
--- a/host/031 - create-gpu-lxc.sh	
+++ b/host/031 - create-gpu-lxc.sh	
@@ -175,6 +175,10 @@ GW_TEMPLATE="10.0.0.1"
 read -r -p "Enter gateway [$GW_TEMPLATE]: " GATEWAY
 GATEWAY=${GATEWAY:-$GW_TEMPLATE}
 
+ROOT_FS_LOCATION_TEMPLATE="local-zfs"
+read -r -p "Enter root fs location [$ROOT_FS_LOCATION_TEMPLATE]: " ROOT_FS_LOCATION
+ROOT_FS_LOCATION=${ROOT_FS_LOCATION:-$ROOT_FS_LOCATION_TEMPLATE}
+
 # Generate random MAC address
 MAC_ADDRESS=$(printf 'BC:24:11:%02X:%02X:%02X\n' $((RANDOM%256)) $((RANDOM%256)) $((RANDOM%256)))
 
@@ -213,7 +217,7 @@ pct create "$CONTAINER_ID" local:vztmpl/ubuntu-24.04-standard_24.04-2_amd64.tar.
     --net0 "name=eth0,bridge=vmbr0,firewall=1,gw=$GATEWAY,hwaddr=$MAC_ADDRESS,ip=$IP_ADDRESS/24,type=veth" \
     --ostype ubuntu \
     --password testing \
-    --rootfs local-zfs:160 \
+    --rootfs $ROOT_FS_LOCATION:160 \
     --swap 4096 \
     --tags "docker;ollama;${ADDITIONAL_TAGS}" \
     --unprivileged 0
diff --git a/lxc/install-docker-and-amd-drivers-in-lxc.sh b/lxc/install-docker-and-amd-drivers-in-lxc.sh
index 44be3aa..47ee9ee 100755
--- a/lxc/install-docker-and-amd-drivers-in-lxc.sh
+++ b/lxc/install-docker-and-amd-drivers-in-lxc.sh
@@ -114,8 +114,8 @@ wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
 
 # Add ROCm 7.1.0 repository (Noble/24.04)
 sudo tee /etc/apt/sources.list.d/rocm.list << EOF
-deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/7.1 noble main
-deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/graphics/7.1/ubuntu noble main
+deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/7.1.1 noble main
+deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/graphics/7.1.1/ubuntu noble main
 EOF
 
 sudo tee /etc/apt/preferences.d/rocm-pin-600 << EOF
@@ -219,4 +219,4 @@ else
     echo -e "${YELLOW}Tests skipped. You can manually test later with:${NC}"
     echo "  docker run --rm --name rcom-smi --device /dev/kfd --device /dev/dri -e HSA_OVERRIDE_GFX_VERSION=11.5.1 -e HSA_ENABLE_SDMA=0 --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ipc=host rocm/rocm-terminal bash -c \"rocm-smi --showmemuse --showuse --showmeminfo all --showhw --showproductname && rocminfo | grep -i -A5 'Agent [0-9]'\""
     echo ""
-fi
\ No newline at end of file
+fi