1 files changed, 2 insertions, 3 deletions
diff --git a/test/test_new.py b/test/test_new.py
index 9ba5f72..c6a8874 100644
--- a/test/test_new.py
+++ b/test/test_new.py
@@ -5,9 +5,8 @@ from mlscraper.parser import make_soup_page, ExtractionResult
 def test_basic():
     html = '<html><body><div class="parent"><p class="item">result</p></div><p class="item">not a result</p></body></html>'
     page = make_soup_page(html)
-    node = page.select(".item")[0]
-    item = {"res": ExtractionResult(node)}
+    item = {"res": "result"}
 
     samples = [SingleItemPageSample(page, item)]
     scraper = RuleBasedSingleItemScraper.build(samples)
-    assert scraper.scrape(html)["res"] == "result"
+    assert scraper.scrape(html) == item