fixed small bug in test