diff --git a/swh/loader/pypi/loader.py b/swh/loader/pypi/loader.py
--- a/swh/loader/pypi/loader.py
+++ b/swh/loader/pypi/loader.py
@@ -125,7 +125,8 @@
         """
         snapshot = self.storage.snapshot_get_latest(self.origin_id)
         if snapshot and snapshot.pop('next_branch', None):
-            return snapshot_get_all_branches(self.storage, snapshot['id'])
+            snapshot = snapshot_get_all_branches(self.storage, snapshot['id'])
+        return snapshot
 
     def prepare(self, project_name, project_url,
                 project_metadata_url=None):