diff --git a/pkg/db/model/node_test.go b/pkg/db/model/node_test.go
index 240e918..7e0a3ce 100644
--- a/pkg/db/model/node_test.go
+++ b/pkg/db/model/node_test.go
@@ -19,9 +19,9 @@
 	"crypto/md5"
 	"fmt"
 	"github.com/golang/protobuf/ptypes/any"
-	"github.com/opencord/voltha-protos/go/common"
-	"github.com/opencord/voltha-protos/go/openflow_13"
-	"github.com/opencord/voltha-protos/go/voltha"
+	"github.com/opencord/voltha-protos/v2/go/common"
+	"github.com/opencord/voltha-protos/v2/go/openflow_13"
+	"github.com/opencord/voltha-protos/v2/go/voltha"
 	"reflect"
 	"testing"
 )