minor change, small bug fix