fix test to improve code coverage