diff --git a/tests/test_nodes/test_nodes.py b/tests/test_nodes/test_nodes.py
index b34c49df..6c1c3ef9 100644
--- a/tests/test_nodes/test_nodes.py
+++ b/tests/test_nodes/test_nodes.py
@@ -377,12 +377,11 @@ def test_functionality_with_separate_node_set(self):
         )
 
         with pytest.raises(BluepySnapError, match="Undefined node set"):
-            self.test_obj.get("ExtraLayer2")
+            next(self.test_obj.get("ExtraLayer2"))
 
-        pdt.assert_frame_equal(
-            self.test_obj.get(node_sets["ExtraLayer2"]),
-            self.test_obj.get("Layer2"),
-        )
+        tested = pd.concat(df for _, df in self.test_obj.get(node_sets["ExtraLayer2"]))
+        expected = pd.concat(df for _, df in self.test_obj.get("Layer2"))
+        pdt.assert_frame_equal(tested, expected)
 
     def test_pickle(self, tmp_path):
         pickle_path = tmp_path / "pickle.pkl"