diff --git a/main/post-install.sh b/main/post-install.sh
index 446fd6a..8813327 100755
--- a/main/post-install.sh
+++ b/main/post-install.sh
@@ -12,6 +12,8 @@ if [[ "-nvidia" == "${NVIDIA_TAG}" ]]; then
     sed -i 's@enabled=1@enabled=0@g' /etc/yum.repos.d/nvidia-container-toolkit.repo
 
     semodule --verbose --install /usr/share/selinux/packages/nvidia-container.pp
+
+    systemctl enable ublue-nvctk-cdi.service
 fi