Update to work with latest changes in Baba