Mend edge case