Edit a small bug