Fixed KDTree import