small bug box fix