Fix generator test