From bd2cfa771e0b31b3749e5caba0ef93feadc19f39 Mon Sep 17 00:00:00 2001
From: David Arnold <dgx.arnold@gmail.com>
Date: Fri, 20 Aug 2021 13:28:48 -0500
Subject: [PATCH] nixos/test: some test fixes in succession of #125992

---
 nixos/tests/common/ec2.nix | 1 +
 1 file changed, 1 insertion(+)

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"