baum welch fix for new implementation, still one block only