From c41101d7647e6564f3b42ef6a23275083c6c75df Mon Sep 17 00:00:00 2001
From: Arkadiusz Szczepkowicz <arek.szczepkowicz@ng-analytics.com>
Date: Fri, 13 Oct 2023 15:10:12 +0200
Subject: [PATCH] #1668: Fix some of the CI issue in tests

---
 src/vt/vrt/collection/manager.impl.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/vt/vrt/collection/manager.impl.h b/src/vt/vrt/collection/manager.impl.h
index e8f4ecb8ca..4a4efa8ff1 100644
--- a/src/vt/vrt/collection/manager.impl.h
+++ b/src/vt/vrt/collection/manager.impl.h
@@ -1683,6 +1683,10 @@ void CollectionManager::insert(
   if (insert_node == this_node and proceed_with_insertion) {
     auto cons_fn = detail::InsertMsgDispatcher<MsgT, ColT>::makeCons(insert_msg);
     makeCollectionElement<ColT>(untyped_proxy, idx, mapped_node, cons_fn);
+
+    auto elm_holder = findElmHolder<IndexType>(untyped_proxy);
+    auto raw_ptr = elm_holder->lookup(idx).getRawPtr();
+    raw_ptr->getLBData().updatePhase(thePhase()->getCurrentPhase());
   } else if (insert_node != this_node) {
     auto msg = makeMessage<InsertMsg<ColT, MsgT>>(
       proxy, idx, insert_node, mapped_node, modify_epoch, insert_msg