Fixes some edge cases