fix test according to pull request comments