updating unit / model tests