Fix generator

@jdoconnor