fixed the tests to use log_probabilities