diff --git a/test_functools.py b/test_functools.py
index ccc192d..6be2d7f 100644
--- a/test_functools.py
+++ b/test_functools.py
@@ -214,7 +214,7 @@ def test_cleanup_called_on_exception(self):
         self.set_to_fail(times=calls)
         retry_call(self.attempt, retries=calls, cleanup=cleanup, trap=Exception)
         assert cleanup.call_count == calls
-        assert cleanup.called_with()
+        cleanup.assert_called_with()
 
     def test_infinite_retries(self):
         self.set_to_fail(times=999)