diff --git a/voltha_omci/omci_frame.py b/voltha_omci/omci_frame.py
index edff530..e05742e 100644
--- a/voltha_omci/omci_frame.py
+++ b/voltha_omci/omci_frame.py
@@ -17,14 +17,14 @@
 from scapy.fields import ShortField, ConditionalField
 from scapy.packet import Packet
 
-from voltha.extensions.omci.omci_defs import FixedLenField
-from voltha.extensions.omci.omci_messages import OmciCreate, OmciDelete, \
+from voltha_omci.omci.omci_defs import FixedLenField
+from voltha_omci.omci.omci_messages import OmciCreate, OmciDelete, \
     OmciDeleteResponse, OmciSet, OmciSetResponse, OmciGet, OmciGetResponse, \
     OmciGetAllAlarms, OmciGetAllAlarmsResponse, OmciGetAllAlarmsNext, \
     OmciMibResetResponse, OmciMibReset, OmciMibUploadNextResponse, \
     OmciMibUploadNext, OmciMibUploadResponse, OmciMibUpload, \
     OmciGetAllAlarmsNextResponse
-from voltha.extensions.omci.omci_messages import OmciCreateResponse
+from voltha_omci.omci.omci_messages import OmciCreateResponse
 
 
 class OmciFrame(Packet):