Cleaning up edge cases