fix probability out