Fix a little bug