Return model type instead of generic object type