diff --git a/.github/workflows/nix_integration_tests.yaml b/.github/workflows/nix_integration_tests.yaml
index 7e8ca8b62..239f56756 100644
--- a/.github/workflows/nix_integration_tests.yaml
+++ b/.github/workflows/nix_integration_tests.yaml
@@ -34,6 +34,7 @@ jobs:
       run: |
         sudo mkdir -p /run/opengl-driver/lib
         sudo find /usr/lib . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
+        sudo find /usr/lib . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
 
         nix develop .#test --command pytest -svv integration-tests/ --release
       env:
diff --git a/flake.nix b/flake.nix
index 45c28637d..674db0d82 100644
--- a/flake.nix
+++ b/flake.nix
@@ -95,7 +95,6 @@
                 cargo
                 rustfmt
                 clippy
-                cudaPackages.cudatoolkit
               ]
               ++ (with python3.pkgs; [
                 docker