diff --git a/python/pyspark/sql/tests/test_sql_context.py b/python/pyspark/sql/tests/test_sql_context.py
index 73a66eb399f5f..e67d28c749580 100644
--- a/python/pyspark/sql/tests/test_sql_context.py
+++ b/python/pyspark/sql/tests/test_sql_context.py
@@ -18,8 +18,7 @@
 import tempfile
 import warnings
 
-from pyspark import SQLContext
-from pyspark.sql import SparkSession
+from pyspark.sql import SparkSession, SQLContext
 from pyspark.testing.sqlutils import ReusedSQLTestCase