Fix wombat feature to comply with new api.