diff --git a/test/yagna/module/payments/test_payment_driver_list_cmd.py b/test/yagna/module/payments/test_payment_driver_list_cmd.py
index 7306980a..cc36051f 100644
--- a/test/yagna/module/payments/test_payment_driver_list_cmd.py
+++ b/test/yagna/module/payments/test_payment_driver_list_cmd.py
@@ -12,10 +12,10 @@
 )
 from goth.node import node_environment
 from goth.runner import Runner
+from goth.runner.container.build import YagnaBuildEnvironment
 from goth.runner.container.payment import PaymentIdPool
 from goth.runner.container.yagna import YagnaContainerConfig
-from goth.runner.requestor import RequestorProbeWithApiSteps
-from goth.runner.container.build import YagnaBuildEnvironment
+from goth.runner.probe import RequestorProbe
 
 
 logger = logging.getLogger(__name__)
@@ -53,7 +53,7 @@ def _topology(payment_id_pool: PaymentIdPool) -> List[YagnaContainerConfig]:
     return [
         YagnaContainerConfig(
             name="requestor",
-            probe_type=RequestorProbeWithApiSteps,
+            probe_type=RequestorProbe,
             environment=requestor_env,
             payment_id=payment_id_pool.get_id(),
         ),
@@ -73,7 +73,7 @@ async def test_payment_driver_list(
     topology = _topology(payment_id_pool)
 
     async with runner(topology):
-        requestor = runner.get_probes(probe_type=RequestorProbeWithApiSteps)[0]
+        requestor = runner.get_probes(probe_type=RequestorProbe)[0]
 
         res = requestor.cli.payment_drivers()
         assert res and res.items()