Fix generator dependency