Update for code coverage integration