diff --git a/examples/src/main/python/sql.py b/examples/src/main/python/sql.py
index 47202fde7510bb7e3975cc862fa499ba04149db0..d89361f324917e751b53fb020ad6e18caeb62b28 100644
--- a/examples/src/main/python/sql.py
+++ b/examples/src/main/python/sql.py
@@ -19,7 +19,7 @@ import os
 
 from pyspark import SparkContext
 from pyspark.sql import SQLContext
-from pyspark.sql import Row, StructField, StructType, StringType, IntegerType
+from pyspark.sql.types import Row, StructField, StructType, StringType, IntegerType
 
 
 if __name__ == "__main__":