Fix a potential bug