small bug fixation