diff --git a/reboot.go b/reboot.go
index f2983ca..2ab3db5 100644
--- a/reboot.go
+++ b/reboot.go
@@ -22,7 +22,7 @@
 	"errors"
 	"fmt"
 	"github.com/google/gopacket"
-	me "github.com/opencord/omci-lib-go/generated"
+	me "github.com/opencord/omci-lib-go/v2/generated"
 )
 
 type RebootRequest struct {