Fix model generation
0.1.4-4