Fixed an issue with overriding a parent class