diff --git a/python/pyspark/streaming/flume.py b/python/pyspark/streaming/flume.py
index 4311e5f5c82aa..8808d60aca5b7 100644
--- a/python/pyspark/streaming/flume.py
+++ b/python/pyspark/streaming/flume.py
@@ -15,7 +15,10 @@
 # limitations under the License.
 #
 
-from StringIO import StringIO
+if sys.version >= "3":
+    from io import StringIO
+else:
+    from StringIO import StringIO
 from py4j.java_gateway import Py4JJavaError
 
 from pyspark.storagelevel import StorageLevel