Handle some edge cases