diff --git a/python/pyspark/java_gateway.py b/python/pyspark/java_gateway.py
index 191dfce1c0460..4e3ad7324d9e2 100644
--- a/python/pyspark/java_gateway.py
+++ b/python/pyspark/java_gateway.py
@@ -34,6 +34,7 @@
 from py4j.java_gateway import java_import, JavaGateway, GatewayParameters
 from pyspark.find_spark_home import _find_spark_home
 from pyspark.serializers import read_int, write_with_length, UTF8Deserializer
+from pyspark.util import _exception_message
 
 
 def launch_gateway(conf=None):