Count Screw up Mechanic

[Chess Ref]

I'm BU. We have a 2-2 count on batter. My partner calls time and talks to OC about something. Comes back behind catcher and annnounces 3-1. Nobody says anything but I know the count is 2-2. Next pitch player swings and misses and everybody is confused. Except me. Partner goes the book and they got one thing , other book has no count, and he comes to me and I got us a strikeout. So he calls her out and nobody has a problem....
I was of the thinking that you don't go to partner until asked. SO thats what I did. i waited ,he asked, I answered.
Should I have stopped the game when he initially gave the wrong count ?
Or at anytime in the process of him asking the two books ?
Or did I do it right ?