Fix code coverage issue