Fix a small bug