diff --git a/test/unit_tests/utils_test.py b/test/unit_tests/utils_test.py
index 5032396..8ff6b69 100644
--- a/test/unit_tests/utils_test.py
+++ b/test/unit_tests/utils_test.py
@@ -1,6 +1,8 @@
-from funcy import rcompose
+from image_prediction.utils.generic import until
 
 
-def test_rcompose():
-    f = rcompose(lambda x: x ** 2, str, lambda x: x * 2)
-    assert f(3) == "99"
+def test_until():
+    def f(x):
+        return x / 2
+
+    assert until(lambda x, y: x - y == 0, f, 1) == 0