Restore the code coverage stuff