small bug correction