Fixed some edge cases.