diff --git a/src/api_clients/invalid_rest.py b/src/api_clients/invalid_rest.py
index 907219b..3c9339f 100644
--- a/src/api_clients/invalid_rest.py
+++ b/src/api_clients/invalid_rest.py
@@ -31,6 +31,7 @@ def alter_dispersal_data(data):
     choice = random.choice([alter_data_content, alter_metadata, add_random_property])
     choice()
 
+    logger.debug(f"Data for dispersal request has been altered to: {data}")
     return data
 
 
@@ -50,16 +51,18 @@ def alter_get_range_query(query):
     choice = random.choice([swap_range, alter_app_id])
     choice()
 
+    logger.debug(f"Get-range query has been altered to: {query}")
     return query
 
 
-class INVALID_REST(REST):
+class InvalidRest(REST):
     def __init__(self, rest_port):
         super().__init__(rest_port)
 
     def send_dispersal_request(self, data):
         data = alter_dispersal_data(data)
-        return self.rest_call("post", "disperse-data", json.dumps(data))
+        response = self.rest_call("post", "disperse-data", json.dumps(data))
+        return response
 
     def send_get_range(self, query):
         query = alter_get_range_query(query)
diff --git a/src/client/nomos_cli.py b/src/client/nomos_cli.py
index 42c230e..f01da6c 100644
--- a/src/client/nomos_cli.py
+++ b/src/client/nomos_cli.py
@@ -2,8 +2,6 @@ import json
 import os
 import re
 
-from src.api_clients.invalid_rest import INVALID_REST
-from src.api_clients.rest import REST
 from src.data_storage import DS
 from src.libs.common import generate_log_prefix, delay, remove_padding
 from src.libs.custom_logger import get_custom_logger
diff --git a/src/client/proxy_client.py b/src/client/proxy_client.py
index e2198da..77c0b78 100644
--- a/src/client/proxy_client.py
+++ b/src/client/proxy_client.py
@@ -2,7 +2,7 @@ import json
 import os
 import re
 
-from src.api_clients.invalid_rest import INVALID_REST
+from src.api_clients.invalid_rest import InvalidRest
 from src.api_clients.rest import REST
 from src.data_storage import DS
 from src.libs.common import generate_log_prefix, delay, remove_padding
@@ -81,7 +81,7 @@ class ProxyClient:
         self._api = REST(self._tcp_port)
 
     def set_invalid_rest_api(self):
-        self._api = INVALID_REST(self._tcp_port)
+        self._api = InvalidRest(self._tcp_port)
 
     @retry(stop=stop_after_delay(5), wait=wait_fixed(0.1), reraise=True)
     def stop(self):
diff --git a/tests/dos_robustness/test_high_load_dos.py b/tests/dos_robustness/test_high_load_dos.py
index 6dc3eeb..2645c97 100644
--- a/tests/dos_robustness/test_high_load_dos.py
+++ b/tests/dos_robustness/test_high_load_dos.py
@@ -170,7 +170,7 @@ class TestHighLoadDos(StepsDataAvailability):
                 break
 
             dispersal_cl, download_cl = random.choice(self.client_nodes[1::2]), random.choice(self.client_nodes[::2])
-            invalid = random.choice([False])
+            invalid = random.choice([False, True])
 
             delay(0.01)
             try: