Add missing import to fix build
Another generator type