diff --git a/napalm_vyos/vyos.py b/napalm_vyos/vyos.py
index dd98e86..20aee55 100644
--- a/napalm_vyos/vyos.py
+++ b/napalm_vyos/vyos.py
@@ -30,10 +30,10 @@
 from netmiko import SCPConn
 
 # NAPALM base
-import napalm_base.constants as C
-from napalm_base.utils import py23_compat
-from napalm_base.base import NetworkDriver
-from napalm_base.exceptions import ConnectionException, \
+import napalm.base.constants as C
+from napalm.base.utils import py23_compat
+from napalm.base.base import NetworkDriver
+from napalm.base.exceptions import ConnectionException, \
                                    MergeConfigException, ReplaceConfigException