Edge case fixes