Fix missing stream in Generator.run