diff --git a/services/maintenance/scripts/titan_24_docker.sh b/services/maintenance/scripts/titan_24_docker.sh
index b5652d52..14353cd2 100644
--- a/services/maintenance/scripts/titan_24_docker.sh
+++ b/services/maintenance/scripts/titan_24_docker.sh
@@ -34,8 +34,12 @@ configure_docker() {
 }
 JSON
   fi
+  if [ -x /host/usr/bin/nvidia-ctk ]; then
+    host_sh "nvidia-ctk runtime configure --runtime=docker"
+  fi
   host_systemctl daemon-reload
   host_systemctl enable --now docker
+  host_systemctl restart docker
 }
 
 if [ ! -x /host/usr/bin/docker ] || ! host_sh "docker version >/dev/null 2>&1"; then