fix model2/Generator 2