Model dataset fix