You can do it this way, but then each faction needs to be given one more hero and a special unit. It's better to divide it into 2 classes instead of 3, plus companion classes are possible (a hero who is a Void Mage or a Rogue).
Added 3 minutes later
+ companion classes are possible (a hero who is a Void Mage or a Rogue)