Update the comparison logic