Fix get_contour function