Fix edge case failure