diff --git a/hack/build/firmware.sh b/hack/build/firmware.sh
index b4e1496..15d80c1 100644
--- a/hack/build/firmware.sh
+++ b/hack/build/firmware.sh
@@ -258,13 +258,13 @@ if [ "${KERNEL_FLAVOR}" = "zone-nvidiagpu" ] && [ "${TARGET_ARCH_STANDARD}" = "x
 	cat > "$ADDONS_OUTPUT_PATH/hooks/nvidia-persist.toml" <<-EOF
 [[hooks.setup]]
 overlay = "nvidia-bootstrap"
-modules = ["nvidia", "nvidia_drm"]
+modules = ["nvidia", "nvidia_drm", "nvidia_uvm"]
 execute = ["/usr/bin/nvidia-smi", "-pm", "1"]
 ignore-failure = true
 
 [[hooks.hotplug]]
 overlay = "nvidia-bootstrap"
-modules = ["nvidia", "nvidia_drm"]
+modules = ["nvidia", "nvidia_drm", "nvidia_uvm"]
 execute = ["/usr/bin/nvidia-smi", "-pm", "1"]
 ignore-failure = true
 EOF