Fix probability
Fixing indeces