diff --git a/patches/ubuntu/build/install-nvidia-container.sh b/patches/ubuntu/files/install-nvidia-container.sh
similarity index 100%
rename from patches/ubuntu/build/install-nvidia-container.sh
rename to patches/ubuntu/files/install-nvidia-container.sh
diff --git a/patches/ubuntu/build/install-nvidia-driver.sh b/patches/ubuntu/files/install-nvidia-driver.sh
similarity index 100%
rename from patches/ubuntu/build/install-nvidia-driver.sh
rename to patches/ubuntu/files/install-nvidia-driver.sh
diff --git a/patches/ubuntu/templates/ubuntu24-full-x64.pkr.hcl b/patches/ubuntu/templates/ubuntu24-full-x64.pkr.hcl
index ca38f6c..31151e9 100644
--- a/patches/ubuntu/templates/ubuntu24-full-x64.pkr.hcl
+++ b/patches/ubuntu/templates/ubuntu24-full-x64.pkr.hcl
@@ -332,13 +332,13 @@ provisioner "shell" {
 provisioner "shell" {
     environment_vars = ["HELPER_SCRIPTS=${var.helper_script_folder}", "INSTALLER_SCRIPT_FOLDER=${var.installer_script_folder}"]
     execute_command  = "sudo sh -c '{{ .Vars }} {{ .Path }}'"
-    scripts          = ["${path.root}/../scripts/build/install-nvidia-driver.sh"]
+    scripts          = ["${path.root}/../custom/files/install-nvidia-driver.sh"]
   }
 
 provisioner "shell" {
     environment_vars = ["HELPER_SCRIPTS=${var.helper_script_folder}", "INSTALLER_SCRIPT_FOLDER=${var.installer_script_folder}"]
     execute_command  = "sudo sh -c '{{ .Vars }} {{ .Path }}'"
-    scripts          = ["${path.root}/../scripts/build/install-nvidia-container.sh"]
+    scripts          = ["${path.root}/../custom/files/install-nvidia-container.sh"]
   }
 
   // provisioner "shell" {