bug fix

small bug fix