diff --git a/eden/scm/edenscm/mercurial/edenapi_upload.py b/eden/scm/edenscm/mercurial/edenapi_upload.py
index 5088bfcd0b..bb692c265e 100644
--- a/eden/scm/edenscm/mercurial/edenapi_upload.py
+++ b/eden/scm/edenscm/mercurial/edenapi_upload.py
@@ -120,14 +120,14 @@ def _uploadtrees(repo, treesbase):
     try:
         with repo.ui.timesection("http.edenapi.upload_trees"):
             stream, _stats = repo.edenapi.uploadtrees(getreponame(repo), trees)
-            foundindices = {item[INDEX_KEY] for item in stream if item[TOKEN_KEY]}
+            trees = list(stream)
             repo.ui.status(
                 _n(
                     "uploaded %d tree\n",
                     "uploaded %d trees\n",
-                    len(foundindices),
+                    len(trees),
                 )
-                % len(foundindices),
+                % len(trees),
                 component="edenapi",
             )
     except (error.RustError, error.HttpError) as e:
diff --git a/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/client.rs b/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/client.rs
index da2307dbec..f7690a58eb 100644
--- a/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/client.rs
+++ b/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/client.rs
@@ -41,7 +41,6 @@ use edenapi_types::TreeEntry;
 use edenapi_types::UploadSnapshotResponse;
 use edenapi_types::UploadToken;
 use edenapi_types::UploadTokensResponse;
-use edenapi_types::UploadTreeResponse;
 use futures::TryStreamExt;
 use minibytes::Bytes;
 use progress::NullProgressFactory;
@@ -345,7 +344,7 @@ py_class!(pub class client |py| {
             Serde<HgId>,  /* p2 */
             PyBytes,      /* data */
         )>,
-    ) -> PyResult<(TStream<anyhow::Result<Serde<UploadTreeResponse>>>, PyFuture)> {
+    ) -> PyResult<(TStream<anyhow::Result<Serde<UploadToken>>>, PyFuture)> {
         self.inner(py).clone().uploadtrees_py(py, repo, items)
     }
 
diff --git a/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/pyext.rs b/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/pyext.rs
index f0730dcad8..4c02a9aa60 100644
--- a/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/pyext.rs
+++ b/eden/scm/edenscmnative/bindings/modules/pyedenapi/src/pyext.rs
@@ -56,7 +56,6 @@ use edenapi_types::UploadHgChangeset;
 use edenapi_types::UploadSnapshotResponse;
 use edenapi_types::UploadToken;
 use edenapi_types::UploadTokensResponse;
-use edenapi_types::UploadTreeResponse;
 use futures::prelude::*;
 use futures::stream;
 use progress::ProgressBar;
@@ -679,7 +678,7 @@ pub trait EdenApiPyExt: EdenApi {
             Serde<HgId>, /* p2 */
             PyBytes,     /* data */
         )>,
-    ) -> PyResult<(TStream<anyhow::Result<Serde<UploadTreeResponse>>>, PyFuture)> {
+    ) -> PyResult<(TStream<anyhow::Result<Serde<UploadToken>>>, PyFuture)> {
         let items = to_trees_upload_items(py, &items)?;
         let (responses, stats) = py
             .allow_threads(|| {
@@ -691,7 +690,7 @@ pub trait EdenApiPyExt: EdenApi {
             .map_pyerr(py)?
             .map_pyerr(py)?;
 
-        let responses_py = responses.map_ok(Serde).map_err(Into::into);
+        let responses_py = responses.map_ok(|r| Serde(r.token)).map_err(Into::into);
         let stats_py = PyFuture::new(py, stats.map_ok(PyStats))?;
         Ok((responses_py.into(), stats_py))
     }