diff --git a/internal/common/omci/reboot.go b/internal/common/omci/reboot.go
index 0f68e02..b5e1fa0 100644
--- a/internal/common/omci/reboot.go
+++ b/internal/common/omci/reboot.go
@@ -19,8 +19,8 @@
 import (
 	"errors"
 	"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"
 	log "github.com/sirupsen/logrus"
 )