diff --git a/uniflowmatch/utils/geometry.py b/uniflowmatch/utils/geometry.py
index 5ca126d..37c97ee 100644
--- a/uniflowmatch/utils/geometry.py
+++ b/uniflowmatch/utils/geometry.py
@@ -10,6 +10,9 @@
 import numpy as np
 import torch
 
+from uniflowmatch.utils.misc import invalid_to_nans
+from scipy.spatial import KDTree
+
 
 def depthmap_to_camera_frame(depthmap, intrinsics):
     """