small bug fix on tests