diff --git a/test/s3/parquet/test_implicit_directory_fix.py b/test/s3/parquet/test_implicit_directory_fix.py
index 7d2023a5c..9ac8f0346 100755
--- a/test/s3/parquet/test_implicit_directory_fix.py
+++ b/test/s3/parquet/test_implicit_directory_fix.py
@@ -14,6 +14,7 @@ import io
 import logging
 import os
 import sys
+import traceback
 
 import pyarrow as pa
 import pyarrow.dataset as pads