diff --git a/samples/snippets/predict_custom_trained_model_sample_test.py b/samples/snippets/predict_custom_trained_model_sample_test.py
index 9b3a60df2a..b3cc1ddaf6 100644
--- a/samples/snippets/predict_custom_trained_model_sample_test.py
+++ b/samples/snippets/predict_custom_trained_model_sample_test.py
@@ -37,4 +37,4 @@ def test_ucaip_generated_predict_custom_trained_model_sample(capsys):
     )
 
     out, _ = capsys.readouterr()
-    assert "number_value: 1.0" in out
+    assert "1.0" in out
diff --git a/samples/snippets/predict_image_object_detection_sample_test.py b/samples/snippets/predict_image_object_detection_sample_test.py
index c4f4893a87..8c2eb2e99f 100644
--- a/samples/snippets/predict_image_object_detection_sample_test.py
+++ b/samples/snippets/predict_image_object_detection_sample_test.py
@@ -31,4 +31,4 @@ def test_ucaip_generated_predict_image_object_detection_sample(capsys):
     )
 
     out, _ = capsys.readouterr()
-    assert 'string_value: "Salad"' in out
+    assert 'Salad' in out
diff --git a/samples/snippets/predict_tabular_classification_sample_test.py b/samples/snippets/predict_tabular_classification_sample_test.py
index 51eed490bd..bc85447ddb 100644
--- a/samples/snippets/predict_tabular_classification_sample_test.py
+++ b/samples/snippets/predict_tabular_classification_sample_test.py
@@ -35,4 +35,4 @@ def test_ucaip_generated_predict_tabular_classification_sample(capsys):
     )
 
     out, _ = capsys.readouterr()
-    assert 'string_value: "setosa"' in out
+    assert 'setosa' in out