MAINT: be more careful about probabilities that are zero