diff --git a/tests/stress/stress_test.py b/tests/stress/stress_test.py
index 82863ee5c..fe9b8604a 100644
--- a/tests/stress/stress_test.py
+++ b/tests/stress/stress_test.py
@@ -141,7 +141,7 @@ def test_function(self, fn):
 
     def run(self):
         print('Cleaning up stress test files from other runs...')
-        cleanup(self._cl, self._TAG, should_wait=True)
+        self.cleanup()
 
         print('Running stress tests...')
         self._start_heartbeat()