diff --git a/hosts/idols-ai/default.nix b/hosts/idols-ai/default.nix
index 98dbb352..49d0e248 100644
--- a/hosts/idols-ai/default.nix
+++ b/hosts/idols-ai/default.nix
@@ -39,11 +39,11 @@ in {
     modesetting.enable = true;
     powerManagement.enable = true;
   };
-  virtualisation.docker.enableNvidia = true; # for nvidia-docker
 
   hardware.graphics = {
     enable = true;
     # needed by nvidia-docker
+    nvidia-container-toolkit.enable = true;
     enable32Bit = true;
   };