diff --git a/staging/src/k8s.io/apiserver/pkg/util/feature/testing/feature_gate_testing.go b/staging/src/k8s.io/apiserver/pkg/util/feature/testing/feature_gate_testing.go
index 6b0a5fe8d3c..b4afc9ae716 100644
--- a/staging/src/k8s.io/apiserver/pkg/util/feature/testing/feature_gate_testing.go
+++ b/staging/src/k8s.io/apiserver/pkg/util/feature/testing/feature_gate_testing.go
@@ -29,16 +29,16 @@ import (
 // Example use:
 //
 // defer utilfeaturetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.<FeatureName>, true)()
-func SetFeatureGateDuringTest(t *testing.T, gate feature.FeatureGate, f feature.Feature, value bool) func() {
+func SetFeatureGateDuringTest(tb testing.TB, gate feature.FeatureGate, f feature.Feature, value bool) func() {
 	originalValue := gate.Enabled(f)
 
 	if err := gate.(feature.MutableFeatureGate).Set(fmt.Sprintf("%s=%v", f, value)); err != nil {
-		t.Errorf("error setting %s=%v: %v", f, value, err)
+		tb.Errorf("error setting %s=%v: %v", f, value, err)
 	}
 
 	return func() {
 		if err := gate.(feature.MutableFeatureGate).Set(fmt.Sprintf("%s=%v", f, originalValue)); err != nil {
-			t.Errorf("error restoring %s=%v: %v", f, originalValue, err)
+			tb.Errorf("error restoring %s=%v: %v", f, originalValue, err)
 		}
 	}
 }