modify model generator bug.