fix for failed tests: ProbabilityFormatter import