update tests from review.