Fix a minor bug