small bug fixex