Fixed incorrect return from Model in case of missing probabilities.