Fix edge detection