fix code coverage