Fix function generator import