I don't think you can say either decision is the correct one. Leave him in or take him out, either decision can be defended.
I'm in the leave him in camp, because of his performance up to that point. Same reason I thought taking Berrios out was wrong.