diff --git a/internal/common/omci/get_test.go b/internal/common/omci/get_test.go
index cffb80d..7a9c21f 100644
--- a/internal/common/omci/get_test.go
+++ b/internal/common/omci/get_test.go
@@ -20,8 +20,8 @@
 	"encoding/hex"
 	"fmt"
 	"github.com/google/gopacket"
-	"github.com/opencord/omci-lib-go"
-	me "github.com/opencord/omci-lib-go/generated"
+	"github.com/opencord/omci-lib-go/v2"
+	me "github.com/opencord/omci-lib-go/v2/generated"
 	"github.com/opencord/voltha-protos/v5/go/openolt"
 	"gotest.tools/assert"
 	"reflect"