diff --git a/pipdeptree.py b/pipdeptree.py
index f2f839a..7348437 100644
--- a/pipdeptree.py
+++ b/pipdeptree.py
@@ -13,7 +13,11 @@ try:
 except ImportError:
     from ordereddict import OrderedDict
 
-import pip
+try:
+    import pip._internal as pip
+except ImportError:
+    import pip
+
 import pkg_resources
 # inline:
 # from graphviz import backend, Digraph
diff --git a/tests/virtualenvs/pickle_env.py b/tests/virtualenvs/pickle_env.py
index ad8ee69..9ac3402 100755
--- a/tests/virtualenvs/pickle_env.py
+++ b/tests/virtualenvs/pickle_env.py
@@ -6,7 +6,10 @@
 import pickle
 import sys
 
-import pip
+try:
+    import pip._internal as pip
+except ImportError:
+    import pip
 
 
 def main():