fix: convert to probability first