Edge case cleanup