diff --git a/nixos/tests/nvidia-container-toolkit.nix b/nixos/tests/nvidia-container-toolkit.nix
index 78d3751eef4d..aa174a8bd7cd 100644
--- a/nixos/tests/nvidia-container-toolkit.nix
+++ b/nixos/tests/nvidia-container-toolkit.nix
@@ -39,7 +39,7 @@ import ./make-test-python.nix (
         name = "cdi-test";
         tag = "latest";
         config = {
-          Cmd = [ "${testCDIScript}/bin/test-cdi" ];
+          Cmd = [ (lib.getExe testCDIScript) ];
         };
         copyToRoot = (
           with pkgs.dockerTools;