Another small bug fixed