Fix some edge cases.