Minor model method change