Minor fix for probing