Fix bug that's breaking unit-coverage task