diff --git a/tests/test_tensor/test_tensor_3d.py b/tests/test_tensor/test_tensor_3d.py
index 9afea631..a7224eab 100644
--- a/tests/test_tensor/test_tensor_3d.py
+++ b/tests/test_tensor/test_tensor_3d.py
@@ -117,4 +117,8 @@ def test_contract_cyclic(self, cyclicx, cyclicy, cyclicz, mode):
         elif mode == "ctmrg":
             lZ = tn.contract_ctmrg(max_bond=chi, cutoff=0.0)
 
-        assert lZ.history_max_size() < 2**13
+
+        if any((cyclicx, cyclicy, cyclicz)):
+            assert lZ.history_max_size() < 2**15
+        else:
+            assert lZ.history_max_size() < 2**13