Wat je zou kunnen doen, is een paar transistors toevoegen om de detecteren dat één of meerdere balanceercircuits actief zijn, en dan stoppen met laden, of de stroom reduceren tot minder dan de stroom die het balanceer circuit uit de cel trekt.
Ik zou ook een weerstandje toevoegen tussen de collector van de transistor en de REF pin van de TL431, om dat circuit een beetje hysterese te geven.
Als alternatief kun je ook een BMS printje uit China kopen, die het balanceren en bewaken voor je doet. Het lastige is dat je heel goed moet controleren of zo'n Chinees ding echt zijn werk doet, maar toevallig heb ik zelf wat 3S BMS boards uitvoerig getest, en dit was bij mijn tests de enige die echt balanceerde en beveiligde: https://nl.aliexpress.com/item/4000059730827.html