Fix polymorphism usage