Corrects the "ValueError: You cannot nest discrete distributions within mixture distributions" Runs in 4.6s on my laptop (M2 chip)