likely issues with `categorical` in generation when requesting them

So far I don't have a use case for this, but I feel this is not completely contemplated in the code. Hence, pay attention to thios.