diff --git a/nixos/tests/common/ec2.nix b/nixos/tests/common/ec2.nix
index 52d0310ac722..64b0a91ac1f7 100644
--- a/nixos/tests/common/ec2.nix
+++ b/nixos/tests/common/ec2.nix
@@ -23,6 +23,7 @@ with pkgs.lib;
       testScript = ''
         import os
         import subprocess
+        import tempfile
 
         image_dir = os.path.join(
             os.environ.get("TMPDIR", tempfile.gettempdir()), "tmp", "vm-state-machine"