correct small bug