Fix prob bin max index calculation