Fixing up the edge cases