Cherry-picked b04e