diff --git a/go/vt/vttablet/queryservice/wrapped.go b/go/vt/vttablet/queryservice/wrapped.go index aa19f911870..624ace3825b 100644 --- a/go/vt/vttablet/queryservice/wrapped.go +++ b/go/vt/vttablet/queryservice/wrapped.go @@ -84,10 +84,10 @@ func wrapInVT15001(err error, inTx bool) error { } c := vterrors.Code(err) m := err.Error() - if c != vtrpcpb.Code_UNAVAILABLE && !strings.Contains(m, "connection refused") { - return err + if c == vtrpcpb.Code_UNAVAILABLE && strings.Contains(m, "connection refused") { + return vterrors.VT15001(c, m) } - return vterrors.VT15001(c, m) + return err } // wrappedService wraps an existing QueryService with