Fix edge cases