Fix bad generation