Fix edge case bug.