Fix the API usage in loop probability heuristics. It was incorrectly