Fix in Baum-Welch