Complete code coverage from tests