diff --git a/tests/setuid.nix b/tests/setuid.nix
index a293666bd..b3b8fe9f1 100644
--- a/tests/setuid.nix
+++ b/tests/setuid.nix
@@ -2,7 +2,7 @@
 
 { nixpkgs, system, nix }:
 
-with import (nixpkgs + "/nixos/lib/testing.nix") { inherit system; };
+with import (nixpkgs + "/nixos/lib/testing-python.nix") { inherit system; };
 
 makeTest {