put back the probability check