Attempt to fix code coverage