improve code coverage
add code coverage