improve code coverage tests
improve code coverage tests