diff --git a/tests/unit/test_wagtail_hooks.py b/tests/unit/test_wagtail_hooks.py
index 55a2d92..75cf976 100644
--- a/tests/unit/test_wagtail_hooks.py
+++ b/tests/unit/test_wagtail_hooks.py
@@ -59,4 +59,4 @@ def test_page_listing_more_buttons(site, rf, segmented_page):
     SegmentFactory(name='something')
     result = wagtail_hooks.page_listing_more_buttons(page, [])
     items = list(result)
-    assert len(items) == 2
+    assert len(items) == 3