Make code coverage work