Updates script for test in .travis.yml