diff --git a/db/model/revision_test.go b/db/model/revision_test.go
index e7da6a5..5a88ff3 100644
--- a/db/model/revision_test.go
+++ b/db/model/revision_test.go
@@ -16,15 +16,15 @@
 package model
 
 import (
-	"github.com/opencord/voltha/protos/go/voltha"
+	"github.com/opencord/voltha-go/protos/voltha"
 	"testing"
 )
 
 func Test_Revision_01_New(t *testing.T) {
 	branch := &Branch{}
 	data := &voltha.Device{}
-	children := make(map[string][]*Revision)
-	rev := NewRevision(branch, data, children)
+	children := make(map[string][]Revision)
+	rev := NewNonPersistedRevision(branch, data, children)
 
 	t.Logf("New revision created: %+v\n", rev)
 }