oops, small bug fix