Fix kdtree example

It's currently broken.