Fixing edge cases.