You definitely did the right thing.
I had a similar situation with Delphi. There was one particular dog who was unruly and had dog aggression, and whose owner was not in control at all. He trampled over Delphi once, and she wanted to sit on my lap for the rest of the class, which I completely understood and allowed her to do. The trainer has since asked the owner to remove the dog from the class; she is working with them privately. If this hadn't happened, I wouldn't have gone back.
Without the big scary dog, Delphi is the belle of the ball
You did what was right for you and your little one.