Yes, I have said several weeks ago the same thing, about Costa has to calm down when opponents want to provoke. But if you see the whole incident, I think his reaction is still quite acceptable this time, carried on by the tense of the game. He is quite calm, and not doing anything more stupid, like slapping, heading, or punching Zabaleta in the end.