Après quelque recherche et test, il n'y a pas réécriture de la classe comme tu me la présenté.
Donc pour l'instant je vais laisser tel quel.