This is a known regression. You can workaround it by putting the comprehension in a different function. It can of course get complicated if the comprehension depends on the input types but you can get around it most of the time by nesting another generated function in the comprehension.