fix a new small bug