diff --git a/tests/unit/location/test_location.cc b/tests/unit/location/test_location.cc
index 35de5bc773..75e66bee5f 100644
--- a/tests/unit/location/test_location.cc
+++ b/tests/unit/location/test_location.cc
@@ -128,10 +128,9 @@ TEST_F(TestLocation, test_migration_entity) {
 
     if (my_node > 1) {
       vt::theLocMan()->virtual_loc->getLocation(entity, 0, [my_node](vt::NodeType node) {
-        // fmt::print("this node={}: node={}\n", my_node, node);
-        // If the 0 node isn't the good one, it's the expected one because it is also the cached one
-        // A routMsg has been send to the last know node (the 0) one which is guarenty to know the correct node)
-        EXPECT_EQ(0, node);
+        // Edit: the expected node can be either 0 (initial) or 1 (migrated)
+        // The protocol may actually eagerly update other nodes
+        EXPECT_TRUE(node == 0 or node == 1);
       });
     } else {
       vt::theLocMan()->virtual_loc->getLocation(entity, 0, [my_node](vt::NodeType node) {