fixed probability issue