Fixes for the edge cases