Fix learning with unified transition matrixes