Update ModelGenerator.scala

'nother hotfix!