Fix a potential bug.