Update tests for a little more coverage.