Fix missing change in the test setup.