Fixed a bug in transition probs