Fix testing setup.