Edge case bug fix