diff --git a/ftplugin/python/pylint_fn.py b/ftplugin/python/pylint_fn.py
index 3a4d1f8..e9888dc 100644
--- a/ftplugin/python/pylint_fn.py
+++ b/ftplugin/python/pylint_fn.py
@@ -3,7 +3,10 @@ Get pylint oputput on current buffer
 """
 import sys
 import re
-from StringIO import StringIO
+try:
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO
 
 try:
     from pylint import lint