Updated for latest api.