opps, small bug fix