diff --git a/src/deep_neurographs/geometry.py b/src/deep_neurographs/geometry.py
index b13bd16..be07134 100644
--- a/src/deep_neurographs/geometry.py
+++ b/src/deep_neurographs/geometry.py
@@ -312,7 +312,7 @@ def query_ball(kdtree, xyz, radius):
         target coordinate.
 
     """
-    idxs = kdtree.query_ball_point(xyz, radius, return_sorted=True)
+    idxs = kdtree.query_ball_point(xyz, radius)
     return kdtree.data[idxs]