diff --git a/pkg/db/model/base_test.go b/pkg/db/model/base_test.go
index 4f44e45..34e6426 100644
--- a/pkg/db/model/base_test.go
+++ b/pkg/db/model/base_test.go
@@ -17,7 +17,7 @@
 
 import (
 	"github.com/opencord/voltha-lib-go/v2/pkg/log"
-	"github.com/opencord/voltha-protos/go/voltha"
+	"github.com/opencord/voltha-protos/v2/go/voltha"
 	"runtime/debug"
 	"sync"
 )