So you're going to tell him to swing at a pitch that he has a rather low probability of hitting (it was a nasty pitch from Max) and was a ball, instead of checking his swing to the point it was not even close. That was on the umpire to get the call right, not Flores.
Frankly, it was a great take and check swing by Flores. The umpire just royally f***ed it up.