fixing some things in multinomial decoder