Updates to tests.