diff --git a/napalm_panos/panos.py b/napalm_panos/panos.py
index 05b6d5e..aa324b9 100644
--- a/napalm_panos/panos.py
+++ b/napalm_panos/panos.py
@@ -30,7 +30,10 @@
 from napalm_base.exceptions import ConnectionException, ReplaceConfigException,\
                                    MergeConfigException
 
-from napalm_base.base import NetworkDriver
+try:
+    from napalm_base import NetworkDriver
+except ImportError:
+    from napalm_base.base import NetworkDriver
 
 from napalm_base.utils import py23_compat