diff --git a/extract_wheels/lib/namespace_pkgs_test.py b/extract_wheels/lib/namespace_pkgs_test.py
index 056bb4ddf6..195cdd9c77 100644
--- a/extract_wheels/lib/namespace_pkgs_test.py
+++ b/extract_wheels/lib/namespace_pkgs_test.py
@@ -133,8 +133,7 @@ def test_empty_case(self):
 
 class TestaddPkgutilStyleNamespacePkgInit(unittest.TestCase):
     def test_missing_directory_is_created(self):
-        directory = TempDir()
-        missing_directory = pathlib.Path(directory.root()) / "missing_directory"
+        missing_directory = pathlib.Path(TempDir().root(), "missing_directory")
         namespace_pkgs.add_pkgutil_style_namespace_pkg_init(str(missing_directory))
         self.assertTrue(missing_directory.is_dir())