Fix a probable bug