Better handle edge cases