Edge case fix for #6