diff --git a/omci_test.go b/omci_test.go
index 86c9a59..bab1e14 100644
--- a/omci_test.go
+++ b/omci_test.go
@@ -21,8 +21,8 @@
 	"encoding/hex"
 	"fmt"
 	"github.com/google/gopacket"
-	. "github.com/opencord/omci-lib-go"
-	. "github.com/opencord/omci-lib-go/generated"
+	. "github.com/opencord/omci-lib-go/v2"
+	. "github.com/opencord/omci-lib-go/v2/generated"
 	"github.com/stretchr/testify/assert"
 	"strings"
 	"testing"