Fix testing set up