Improved edge case