diff --git a/hosts/canopus/default.nix b/hosts/canopus/default.nix
index f3805ce..d510cf9 100755
--- a/hosts/canopus/default.nix
+++ b/hosts/canopus/default.nix
@@ -21,8 +21,8 @@
     ../../modules/nixos/steam.nix
   ];
 
+  hardware.nvidia-container-toolkit.enable = true;
   tux.services.openssh.enable = true;
-
   nixpkgs.config.cudaSupport = true;
 
   sops.secrets = {
diff --git a/hosts/sirius/default.nix b/hosts/sirius/default.nix
index 1c988bc..b679806 100644
--- a/hosts/sirius/default.nix
+++ b/hosts/sirius/default.nix
@@ -12,7 +12,10 @@
     ../../modules/nixos/virtualisation/docker.nix
   ];
 
-  hardware.nvidia-container-toolkit.suppressNvidiaDriverAssertion = true;
+  hardware.nvidia-container-toolkit = {
+    enable = true;
+    suppressNvidiaDriverAssertion = true;
+  };
 
   tux.services.openssh.enable = true;
 
diff --git a/modules/nixos/virtualisation/docker.nix b/modules/nixos/virtualisation/docker.nix
index 42fc78a..d93f20a 100755
--- a/modules/nixos/virtualisation/docker.nix
+++ b/modules/nixos/virtualisation/docker.nix
@@ -8,8 +8,6 @@
     docker.enable = true;
   };
 
-  hardware.nvidia-container-toolkit.enable = true;
-
   environment.systemPackages = with pkgs; [lazydocker];
 
   users.users.${username}.extraGroups = ["docker"];