Update travis script to get it to work with current setup