[CHANGE] Small change in the setup. Check if still pass in travis