Fix bug: all prob sum to 1