updated base model to work properly with the new collection object