He could do with a break I think but his bowling hasn't been dreadful in the ODIs. Only two wickets in the three matches but also 27 overs have yielded just 131 runs in total, so not giving away much free stuff.
I'd suggest resting him for the games in New Zealand, especially if Stokes is back.
I think they should go back to playing the one-day games before the Tests, which would be a good 'appetiser', and give players the chance to do well and break into the Test team. Feels a bit of a damp squib, with the Ashes gone