diff --git a/swh/graph/tests/test_api_client.py b/swh/graph/tests/test_api_client.py
--- a/swh/graph/tests/test_api_client.py
+++ b/swh/graph/tests/test_api_client.py
@@ -106,9 +106,12 @@
 def test_visit_edges_limited(graph_client):
     actual = list(
         graph_client.visit_edges(
-            "swh:1:rel:0000000000000000000000000000000000000010", max_edges=4
+            "swh:1:rel:0000000000000000000000000000000000000010",
+            max_edges=4,
+            edges="rel:rev,rev:rev,rev:dir",
         )
     )
+    assert len(actual) == 3
     expected = [
         (
             "swh:1:rel:0000000000000000000000000000000000000010",
@@ -116,14 +119,18 @@
         ),
         (
             "swh:1:rev:0000000000000000000000000000000000000009",
-            "swh:1:dir:0000000000000000000000000000000000000008",
+            "swh:1:rev:0000000000000000000000000000000000000003",
         ),
         (
             "swh:1:rev:0000000000000000000000000000000000000009",
+            "swh:1:dir:0000000000000000000000000000000000000008",
+        ),
+        (
             "swh:1:rev:0000000000000000000000000000000000000003",
+            "swh:1:dir:0000000000000000000000000000000000000002",
         ),
     ]
-    assert set(actual) == set(expected)
+    assert set(actual).issubset(set(expected))
 
 
 def test_visit_edges_diamond_pattern(graph_client):