[travis] Improve script