Edge case tweak