Fixes bugs of the multinomial components