Multinomial returns nan

Ah you beat me to it :slight_smile: Thanks!