Update setup for tests.