fix for probability