diff --git a/registrasion/tests/test_cart.py b/registrasion/tests/test_cart.py
index f8a82c21ae8985d4b1908d5d69614480553c671a..066bf3774609e0159aa86b4e9219d66936b4ac42 100644
--- a/registrasion/tests/test_cart.py
+++ b/registrasion/tests/test_cart.py
@@ -95,9 +95,9 @@ class RegistrationCartTestCase(SetTimeMixin, TestCase):
 
     @classmethod
     def make_ceiling(cls, name, limit=None, start_time=None, end_time=None):
-        limit_ceiling = rego.TimeOrStockLimitEnablingCondition.objects.create(
+        limit_ceiling = rego.TimeOrStockLimitFlag.objects.create(
             description=name,
-            mandatory=True,
+            condition=rego.FlagBase.DISABLE_IF_FALSE,
             limit=limit,
             start_time=start_time,
             end_time=end_time
@@ -109,9 +109,9 @@ class RegistrationCartTestCase(SetTimeMixin, TestCase):
     @classmethod
     def make_category_ceiling(
             cls, name, limit=None, start_time=None, end_time=None):
-        limit_ceiling = rego.TimeOrStockLimitEnablingCondition.objects.create(
+        limit_ceiling = rego.TimeOrStockLimitFlag.objects.create(
             description=name,
-            mandatory=True,
+            condition=rego.FlagBase.DISABLE_IF_FALSE,
             limit=limit,
             start_time=start_time,
             end_time=end_time