diff --git a/python/python/tests/test_session_config.py b/python/python/tests/test_session_config.py
index 3ec51dc74c..bac1e3e00e 100644
--- a/python/python/tests/test_session_config.py
+++ b/python/python/tests/test_session_config.py
@@ -225,7 +225,7 @@ def test_set_overrides_ballista_defaults():
         "ballista.shuffle.sort_based.memory_limit_per_task_bytes",
         "1024",
     ),
-    ("with_shuffle_compression_codec", "ballista.shuffle.compression.codec", "zstd"),
+    ("with_shuffle_compression_codec", "ballista.shuffle.compression.codec", "lz4"),
     (
         "with_grpc_client_initial_connection_window_size",
         "ballista.client.initial_connection_window_size",