diff --git a/python/python_test.go b/python/python_test.go
index 3f719f8..bb407e4 100644
--- a/python/python_test.go
+++ b/python/python_test.go
@@ -305,7 +305,7 @@
 
 func TestPythonModule(t *testing.T) {
 	config, buildDir := setupBuildEnv(t)
-	defer tearDownBuildEnv()
+	defer tearDownBuildEnv(buildDir)
 	android.TestPreDepsMutators(func(ctx android.RegisterMutatorsContext) {
 		ctx.BottomUp("version_split", versionSplitMutator()).Parallel()
 	})
@@ -432,8 +432,8 @@
 	return
 }
 
-func tearDownBuildEnv() {
-	os.RemoveAll(buildNamePrefix)
+func tearDownBuildEnv(buildDir string) {
+	os.RemoveAll(buildDir)
 }
 
 func findModule(ctx *blueprint.Context, name, variant string) blueprint.Module {