Compatibility for yelp-cheetah==0.4.0

Only changed the xml, the inputs / outputs are still the same.