Fix testing setup