diff --git a/python/python_test.go b/python/python_test.go
index 0e97626..5f971cd 100644
--- a/python/python_test.go
+++ b/python/python_test.go
@@ -313,7 +313,7 @@
 }
 
 func expectModule(t *testing.T, ctx *android.TestContext, name, variant, expectedSrcsZip string, expectedPyRunfiles []string) {
-	module := ctx.ModuleForTests(name, variant)
+	module := ctx.ModuleForTests(t, name, variant)
 
 	base, baseOk := module.Module().(*PythonLibraryModule)
 	if !baseOk {