diff --git a/gsrp_smart_util.py b/gsrp_smart_util.py
index 0e42e8d5a76e63f60d961d6d71f37bfee484d0b9..8dfb6a2378f41a8cc31ee61534f8c22e7311d00e 100644
--- a/gsrp_smart_util.py
+++ b/gsrp_smart_util.py
@@ -127,8 +127,8 @@ def add(mem1, mem2, cc, t_max, id1, id2, n_ind, mem_limit=np.infty):
     return out_val, out_tij
 
 
-def mask_val(mem, val):
-    mask = mem[0] >= val
+def mask_val(mem, val, atol=5e-9):
+    mask = mem[0] >= val - atol
     return mem[0][mask], np.compress(mask, mem[1], axis=1)