diff --git a/internal/pkg/afrouter/method-router_test.go b/internal/pkg/afrouter/method-router_test.go
index fd0285a..3f7d12c 100644
--- a/internal/pkg/afrouter/method-router_test.go
+++ b/internal/pkg/afrouter/method-router_test.go
@@ -19,8 +19,8 @@
 import (
 	"fmt"
 	"github.com/golang/protobuf/proto"
-	common_pb "github.com/opencord/voltha-protos/go/common"
-	voltha_pb "github.com/opencord/voltha-protos/go/voltha"
+	common_pb "github.com/opencord/voltha-protos/v2/go/common"
+	voltha_pb "github.com/opencord/voltha-protos/v2/go/voltha"
 	"github.com/stretchr/testify/assert"
 	"google.golang.org/grpc"
 	"testing"