diff --git a/flake8_bandit.py b/flake8_bandit.py
index e76d534..7f4a88c 100644
--- a/flake8_bandit.py
+++ b/flake8_bandit.py
@@ -15,8 +15,13 @@
 except ImportError:
     import ConfigParser as configparser
 
+try:
+    from flake8.engine import pep8 as stdin_utils
+except ImportError:
+    from flake8 import utils as stdin_utils
+
 
-__version__ = "2.1.1"
+__version__ = "2.1.2"
 
 
 class BanditTester(object):
@@ -90,7 +95,7 @@ def _load_source(self):
 
         if self.filename in ("stdin", "-", None):
             self.filename = "stdin"
-            self.lines = pycodestyle.stdin_get_value().splitlines(True)
+            self.lines = stdin_utils.stdin_get_value().splitlines(True)
         else:
             self.lines = pycodestyle.readlines(self.filename)
         if not self.tree: