Restoring code coverage