fix code coverage issue