However, I think there's something massively wrong with a system that punishes success. What might need to be different to prevent that sort of outcome?

Letting founders remain permanently in control under all forms of incorporation is very far from profit-maximizing, sorry! The system is working as designed.