TST: Fixed issue with travis-ci and new test script