diff --git a/fs/bcachefs/recovery.c b/fs/bcachefs/recovery.c
index 97c0d7d1fe77..095eef3828ce 100644
--- a/fs/bcachefs/recovery.c
+++ b/fs/bcachefs/recovery.c
@@ -268,10 +268,12 @@ static int bch2_extent_replay_key(struct bch_fs *c, enum btree_id btree_id,
 				   BTREE_ITER_INTENT);
 
 	do {
-		ret = bch2_extent_atomic_end(iter, k, &atomic_end);
+		ret = bch2_btree_iter_traverse(iter);
 		if (ret)
 			goto err;
 
+		atomic_end = bpos_min(k->k.p, iter->l[0].b->key.k.p);
+
 		split_iter = bch2_trans_copy_iter(&trans, iter);
 		ret = PTR_ERR_OR_ZERO(split_iter);
 		if (ret)