Fix script for running test on Travis CI