fix probabilistic model