diff --git a/dataproc/dataproc_e2e_test.py b/dataproc/dataproc_e2e_test.py
index c624b8c8d39..56db9b1146b 100644
--- a/dataproc/dataproc_e2e_test.py
+++ b/dataproc/dataproc_e2e_test.py
@@ -27,4 +27,4 @@
 def test_e2e(cloud_config):
     output = create_cluster_and_submit_job.main(
         cloud_config.project, ZONE, CLUSTER_NAME, cloud_config.storage_bucket)
-    assert "['Hello,', 'dog', 'elephant', 'panther', 'world!']" in output
+    assert b"['Hello,', 'dog', 'elephant', 'panther', 'world!']" in output