Fix a small mistake