fix to reflect changes made in the ModelGenerator class