Fixing bug with updating the view when doing collection remove