Update to use new version of Juju API