fix issue with generated models