Fix generate bug