small bug fix, testing