small bug update