Fix test setup.