diff --git a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/units/nvidia.service b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/units/nvidia.service
index 1b36dfe873c..66575b65824 100644
--- a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/units/nvidia.service
+++ b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/units/nvidia.service
@@ -2,9 +2,11 @@
 Description=NVIDIA Configure Service
 Wants=network-online.target
 After=network-online.target
+Before=containerd.target
 
 [Service]
-Type=simple
+Type=oneshot
+RemainsAfterExit=true
 Restart=no
 Environment=PATH=/opt/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin
 ExecStart=/usr/lib/nvidia/bin/setup-nvidia