Derp
Add better distance check