diff --git a/netbox_healthcheck_plugin/__init__.py b/netbox_healthcheck_plugin/__init__.py
index 89b0c20..fff25d7 100644
--- a/netbox_healthcheck_plugin/__init__.py
+++ b/netbox_healthcheck_plugin/__init__.py
@@ -5,7 +5,7 @@
 __version__ = '0.1.0'
 
 
-from extras.plugins import PluginConfig
+from netbox.plugins import PluginConfig
 
 
 class HealthCheckConfig(PluginConfig):
diff --git a/netbox_healthcheck_plugin/navigation.py b/netbox_healthcheck_plugin/navigation.py
index 83db1d9..6979386 100644
--- a/netbox_healthcheck_plugin/navigation.py
+++ b/netbox_healthcheck_plugin/navigation.py
@@ -1,5 +1,4 @@
-from extras.plugins import PluginMenuButton, PluginMenuItem
-from utilities.choices import ButtonColorChoices
+from netbox.plugins import PluginMenuButton, PluginMenuItem
 
 
 menu_items = (