Tests: update script for Travis CI