diff --git a/codes/tb/tb.py b/codes/tb/tb.py
index c943d405a95408353f8f392d5c18c5e758eed4a8..3a178527297842cae22beddaaf2a7f4e9e395b59 100644
--- a/codes/tb/tb.py
+++ b/codes/tb/tb.py
@@ -20,6 +20,6 @@ def add_tb(tb1, tb2):
     return {k: tb1.get(k, 0) + tb2.get(k, 0) for k in frozenset(tb1) | frozenset(tb2)}
 
 
-def compare_dicts(dict1, dict2):
+def compare_dicts(dict1, dict2, atol=1e-10):
     for key in dict1.keys():
-        assert np.allclose(dict1[key], dict2[key])
+        assert np.allclose(dict1[key], dict2[key], atol=atol)