oops, small bug fix.