diff --git a/PyCTBN/tests/structure_graph/test_cim.py b/PyCTBN/tests/structure_graph/test_cim.py
index f26d9b8..7106974 100644
--- a/PyCTBN/tests/structure_graph/test_cim.py
+++ b/PyCTBN/tests/structure_graph/test_cim.py
@@ -32,7 +32,7 @@ class TestConditionalIntensityMatrix(unittest.TestCase):
                 c2[i, j] = (c2[i, j] + 1) / (self.state_res_times[i] + 1)
         c1.compute_cim_coefficients()
         for i in range(0, len(c1.state_residence_times)):
-            self.assertTrue(np.isclose(np.sum(c1.cim[i]), 0.0, 1e01))
+            self.assertTrue(np.isclose(np.sum(c1.cim[i]), 0.0, 1e01,1e01))
         for i in range(0, len(self.state_res_times)):
             for j in range(0, len(self.state_res_times)):
                 self.assertTrue(np.isclose(c1.cim[i, j], c2[i, j], 1e01))