diff --git a/voltha_omci/omci_messages.py b/voltha_omci/omci_messages.py
index 90a0cea..0c78297 100644
--- a/voltha_omci/omci_messages.py
+++ b/voltha_omci/omci_messages.py
@@ -19,8 +19,8 @@
 from scapy.fields import ShortField
 from scapy.packet import Packet
 
-from voltha.extensions.omci.omci_defs import AttributeAccess
-from voltha.extensions.omci.omci_entities import entity_id_to_class_map
+from voltha_omci.omci.omci_defs import AttributeAccess
+from voltha_omci.omci.omci_entities import entity_id_to_class_map
 
 
 log = structlog.get_logger()