Minor fix in KDtree