diff --git a/setup.py b/setup.py
index 67b144f..6e786f8 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 
 install_requires = [
     "tqdm>=4.15.0,<5",
-    "wagtail>=2.15,<4",
+    "wagtail>=2.15,<3",
 ]
 
 
diff --git a/wagtailinventory/tests/test_helpers.py b/wagtailinventory/tests/test_helpers.py
index e83ded1..32aa00e 100644
--- a/wagtailinventory/tests/test_helpers.py
+++ b/wagtailinventory/tests/test_helpers.py
@@ -1,15 +1,11 @@
 from django.test import TestCase
 
-import wagtail
 from wagtail.core.models import Page
 
 from wagtailinventory.helpers import get_page_blocks
 
 
-if wagtail.VERSION < (3,):
-    CORE_BLOCKS = "wagtail.core.blocks"
-else:
-    CORE_BLOCKS = "wagtail.blocks"
+CORE_BLOCKS = "wagtail.core.blocks"
 
 
 class TestGetPageBlocks(TestCase):