Removed type hinting for models in base class