edge case bug fix