Fix compilation of ProbingPT