improve code coverage

improve code coverage