diff --git a/internal/test/mock/grpc_testify_mock.go b/internal/test/mock/grpc_testify_mock.go
index c853f27d4e..cca58973f4 100644
--- a/internal/test/mock/grpc_testify_mock.go
+++ b/internal/test/mock/grpc_testify_mock.go
@@ -213,3 +213,5 @@ func (c *ClientInternal) Close(ctx context.Context) error {
 	args := c.Called(ctx)
 	return args.Error(0)
 }
+
+func (c *ClientInternal) SetDisableResolveDNSAddr(addr string, distributed bool) {}