diff --git a/tests/benchmark/locustfile.py b/tests/benchmark/locustfile.py
index 2769150c..d8009c82 100644
--- a/tests/benchmark/locustfile.py
+++ b/tests/benchmark/locustfile.py
@@ -24,7 +24,7 @@
 
 import os
 
-from locust import HttpLocust, TaskSet, between, task
+from locust import HttpUser, TaskSet, between, task
 
 TOKEN = os.environ.get("REANA_ACCESS_TOKEN")
 
@@ -80,8 +80,8 @@ def get_worflow_logs(self):
         )
 
 
-class WebsiteUser(HttpLocust):
+class WebsiteUser(HttpUser):
     """Locust instance. Represent the user that attacks the system."""
 
-    task_set = WorkflowsTaskSet
+    tasks = [WorkflowsTaskSet]
     wait_time = between(5, 10)