Minor change at model