Refactor tests after review changes.