fix code coverage problem