update, small bug fix