Fixed code coverage failure