I agree with the others. Part of this process, if not most of this process is about healing ourselves and making ourselves whole. Calling your W out on unacceptable behavior is a good thing, particularly when the motivation is to show your child how to be.
One of the few things we control in this mess is our own feelings (I know I am the poster child for crying and feeling low, but the fact of the matter is, that it is almostentirely my own doing). I think it was great that you took care of your needs in that situation.