Set up tests to work with Travis CL