Edge case fix