TST: Changing tests to get it to work in travis