diff --git a/nixos/lib/test-driver/shell.nix b/nixos/lib/test-driver/shell.nix
index 8879b1a0dc4c..9f3e34868be2 100644
--- a/nixos/lib/test-driver/shell.nix
+++ b/nixos/lib/test-driver/shell.nix
@@ -1,4 +1,4 @@
 {
   pkgs ? import ../../.. { },
 }:
-pkgs.callPackage ./default.nix { }
+pkgs.python3Packages.callPackage ./default.nix { }