Fix generator integration test