Fix a small bug,