diff --git a/plugin/grpc_provider_test.go b/plugin/grpc_provider_test.go
index a2fe37a3ab..cacc36a2e1 100644
--- a/plugin/grpc_provider_test.go
+++ b/plugin/grpc_provider_test.go
@@ -26,6 +26,7 @@ func mockProviderClient(t *testing.T) *mockproto.MockProviderClient {
 	client.EXPECT().GetSchema(
 		gomock.Any(),
 		gomock.Any(),
+		gomock.Any(),
 	).Return(providerProtoSchema(), nil)
 
 	return client