Use get_class and get_model to allow overriding