Fix edge case