Update tests.

100% code coverage.