Add tests for uncovered code.