Update for changes to Stage api