diff --git a/tests/test_helpers.py b/tests/test_helpers.py
index 6fedbb85c..3f8d8246b 100644
--- a/tests/test_helpers.py
+++ b/tests/test_helpers.py
@@ -6,7 +6,9 @@
 import requests_mock
 
 from regolith.main import main
-from regolith.schemas import MILESTONE_TYPES
+from regolith.schemas import alloweds
+
+MILESTONE_TYPES = alloweds.get("MILESTONE_TYPES")
 
 dash = "-"
 helper_map = [