fix for generator