diff --git a/openfeature/testing/testprovider.go b/openfeature/testing/testprovider.go
index fb42d3a..584b993 100644
--- a/openfeature/testing/testprovider.go
+++ b/openfeature/testing/testprovider.go
@@ -29,7 +29,7 @@ type TestProvider struct {
 }
 
 // UsingFlags sets flags for the scope of a test
-func (tp TestProvider) UsingFlags(test *testing.T, flags map[string]memprovider.InMemoryFlag) {
+func (tp TestProvider) UsingFlags(test testing.TB, flags map[string]memprovider.InMemoryFlag) {
 	storeGoroutineLocal(test.Name())
 	tp.providers.Store(test.Name(), memprovider.NewInMemoryProvider(flags))
 }