fix transition probability expression