From 9d857bb3c44cbe12005e73f37750f239b68953b4 Mon Sep 17 00:00:00 2001
From: Evan Morris <evandietzmorris@gmail.com>
Date: Tue, 27 Aug 2024 00:24:54 -0400
Subject: [PATCH] fixed for new cypher results format

---
 tests/test_subclass.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_subclass.py b/tests/test_subclass.py
index d0cf451..7d5c842 100644
--- a/tests/test_subclass.py
+++ b/tests/test_subclass.py
@@ -65,7 +65,7 @@ def test_onehop_subclass_categories():
     }
     query = get_query(qgraph)
     #make sure that the class (PhenotypicFeature) has been removed from n0
-    clause = query.split('RETURN')[0]
+    clause = query.split('WITH')[0]
     elements = clause.split('-')
     checked = False
     for element in elements: