diff --git a/modules/base.nix b/modules/base.nix
index 41e8333..0224280 100644
--- a/modules/base.nix
+++ b/modules/base.nix
@@ -75,7 +75,6 @@ with lib;
 
   virtualisation.docker.enable = true;
   # virtualisation.containers.cdi.dynamic.nvidia.enable = true;
-  hardware.nvidia-container-toolkit.enable = true ;
   services.openssh = {
     enable = true;
     settings.PermitRootLogin = "yes";
diff --git a/system/shen/configuration.nix b/system/shen/configuration.nix
index 1894569..8201ef5 100644
--- a/system/shen/configuration.nix
+++ b/system/shen/configuration.nix
@@ -30,6 +30,7 @@ in
 
   networking.hostName = "shen"; # Define your hostname.
 
+  hardware.nvidia-container-toolkit.enable = true ;
   # Enable CUPS to print documents.
   services.printing = {
     enable = true;