Fix build error introduced for add code coverage