A small bug fixing