edge case fix