Ah, so the Categorical cannot be adjusted to work, as it will have a non-zero probability of having a value of 0, which is not a possible value for n. Or will will have a zero probability of having a value of 0, which would be -inf logp? Causing the warning?