Fix a small bug , never mind