fix generate bug