Update for new Wes API