Fix generate command