slight model change