diff --git a/internal/common/omci/set_test.go b/internal/common/omci/set_test.go
index 33ed35a..d896e53 100644
--- a/internal/common/omci/set_test.go
+++ b/internal/common/omci/set_test.go
@@ -17,8 +17,8 @@
 package omci
 
 import (
-	"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/stretchr/testify/assert"
 	"testing"
 )