diff --git a/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php b/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php
index eb96742acf3993f830da9305d1ba10b827bd8ba8..39d12666ad994d5685684180428943fe47463e63 100644
--- a/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php
+++ b/tests/src/Unit/LinkManipulator/ParseLinkSampleTest.php
@@ -28,11 +28,14 @@ class ParseLinkSampleTest extends UnitTestCase {
     if ($request_context['sample']) {
       $this->assertEmpty($request_context['entity_ids']);
     }
-    elseif (count($entity_ids) == 2) {
-      $this->assertEquals(['node' => ['1' => '1', '2' => '2']], $request_context['entity_ids']);
-    }
     else {
-      $this->assertEquals(['node' => ['1' => '1']], $request_context['entity_ids']);
+      $expected = [
+        'node' => ['1' => '1'],
+      ];
+      if (is_array($entity_ids) && count($entity_ids) == 2) {
+        $expected['node']['2'] = '2';
+      }
+      $this->assertEquals($expected, $request_context['entity_ids']);
     }
   }