Update expectations for new graph API