diff --git a/pkg/db/model/transaction_test.go b/pkg/db/model/transaction_test.go
index 3660a86..c33e5be 100644
--- a/pkg/db/model/transaction_test.go
+++ b/pkg/db/model/transaction_test.go
@@ -20,8 +20,8 @@
 	"context"
 	"encoding/hex"
 	"github.com/google/uuid"
-	"github.com/opencord/voltha-protos/go/common"
-	"github.com/opencord/voltha-protos/go/voltha"
+	"github.com/opencord/voltha-protos/v2/go/common"
+	"github.com/opencord/voltha-protos/v2/go/voltha"
 	"strconv"
 	"testing"
 )