Fix the script used for Travis