I had a similar situation, I waited until we were away from the situation and then commented "I think it was really wrong the way you were spoken to by X".