A small bug fixed