fixing code-coverage