diff --git a/clrs/_src/losses_test.py b/clrs/_src/losses_test.py
index 44a181fa..7a1e50e6 100644
--- a/clrs/_src/losses_test.py
+++ b/clrs/_src/losses_test.py
@@ -159,7 +159,7 @@ def test_hint_loss(self, algo):
           lengths=full_sample.features.lengths,
           nb_nodes=nb_nodes,
       )
-      np.testing.assert_allclose(chunk_hint_loss, full_hint_loss, rtol=1e-4)
+      np.testing.assert_allclose(chunk_hint_loss, full_hint_loss, rtol=1e-3)
 
 
 if __name__ == '__main__':