diff --git a/sdk/ahnlich-client-py/ahnlich_client_py/tests/ai_client/test_ai_client_store_commands.py b/sdk/ahnlich-client-py/ahnlich_client_py/tests/ai_client/test_ai_client_store_commands.py
index 21937ab7..4342f51b 100644
--- a/sdk/ahnlich-client-py/ahnlich_client_py/tests/ai_client/test_ai_client_store_commands.py
+++ b/sdk/ahnlich-client-py/ahnlich_client_py/tests/ai_client/test_ai_client_store_commands.py
@@ -53,9 +53,7 @@ def test_ai_client_get_pred(spin_up_ahnlich_ai):
     builder.set(
         store_name=ai_store_payload_with_predicates["store_name"],
         inputs=store_inputs,
-        preprocess_action=ai_query.PreprocessAction__RawString(
-            ai_query.StringAction__ErrorIfTokensExceed()
-        ),
+        preprocess_action=ai_query.PreprocessAction__NoPreprocessing(),
     )
     expected = ai_response.AIServerResult(
         results=[
@@ -196,9 +194,7 @@ def test_ai_client_del_key(spin_up_ahnlich_ai):
     builder.set(
         store_name=ai_store_payload_with_predicates["store_name"],
         inputs=store_inputs,
-        preprocess_action=ai_query.PreprocessAction__RawString(
-            ai_query.StringAction__ErrorIfTokensExceed()
-        ),
+        preprocess_action=ai_query.PreprocessAction__NoPreprocessing(),
     )
     expected = ai_response.AIServerResult(
         results=[