diff --git a/modules/nvidia.nix b/modules/nvidia.nix
index 093bb7c..9ec7931 100644
--- a/modules/nvidia.nix
+++ b/modules/nvidia.nix
@@ -22,6 +22,7 @@
     nvidia-container-toolkit
   ];
 
+  services.k3s.nodeLabel = ["nixos-nvidia-cdi=enabled"];
   services.k3s.containerdConfigTemplate = ''
     {{ template "base" . }}
 
@@ -30,8 +31,5 @@
       runtime_engine = ""
       runtime_root = ""
       runtime_type = "io.containerd.runc.v2"
-
-    [plugins."io.containerd.grpc.v1.cri".containerd.runtimes.nvidia.options]
-      BinaryName = "${lib.getOutput "tools" config.hardware.nvidia-container-toolkit.package}/bin/nvidia-container-runtime"
   '';
 }