diff --git a/testing/test_spec.go b/testing/test_spec.go
index 2a05456..02f6c60 100644
--- a/testing/test_spec.go
+++ b/testing/test_spec.go
@@ -119,7 +119,7 @@
 	}
 	android.WriteFileRule(ctx, intermediatePath, string(protoData))
 
-	ctx.SetProvider(
+	android.SetProvider(ctx,
 		TestSpecProviderKey, TestSpecProviderData{
 			IntermediatePath: intermediatePath,
 		},