diff --git a/src/common/models/notification.go b/src/pkg/p2p/preheat/models/notification/notification.go
similarity index 98%
rename from src/common/models/notification.go
rename to src/pkg/p2p/preheat/models/notification/notification.go
index ef4cba067..2d9e7fe1d 100644
--- a/src/common/models/notification.go
+++ b/src/pkg/p2p/preheat/models/notification/notification.go
@@ -12,7 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-package models
+package notification
 
 import (
 	"time"
diff --git a/src/pkg/p2p/preheat/provider/kraken.go b/src/pkg/p2p/preheat/provider/kraken.go
index 760a46cfc..3396fce3c 100644
--- a/src/pkg/p2p/preheat/provider/kraken.go
+++ b/src/pkg/p2p/preheat/provider/kraken.go
@@ -3,11 +3,11 @@ package provider
 import (
 	"errors"
 	"fmt"
+	"github.com/goharbor/harbor/src/pkg/p2p/preheat/models/notification"
 	"strings"
 	"time"
 
 	"github.com/docker/distribution/manifest/schema2"
-	cm "github.com/goharbor/harbor/src/common/models"
 	"github.com/goharbor/harbor/src/common/utils"
 	"github.com/goharbor/harbor/src/pkg/p2p/preheat/models/provider"
 	"github.com/goharbor/harbor/src/pkg/p2p/preheat/provider/auth"
@@ -67,13 +67,13 @@ func (kd *KrakenDriver) Preheat(preheatingImage *PreheatImage) (*PreheatingStatu
 	}
 
 	url := fmt.Sprintf("%s%s", strings.TrimSuffix(kd.instance.Endpoint, "/"), krakenPreheatPath)
-	var events = make([]cm.Event, 0)
+	var events = make([]notification.Event, 0)
 	eventID := utils.GenerateRandomString()
-	event := cm.Event{
+	event := notification.Event{
 		ID:        eventID,
 		TimeStamp: time.Now().UTC(),
 		Action:    "push",
-		Target: &cm.Target{
+		Target: &notification.Target{
 			MediaType:  schema2.MediaTypeManifest,
 			Digest:     preheatingImage.Digest,
 			Repository: preheatingImage.ImageName,
@@ -82,7 +82,7 @@ func (kd *KrakenDriver) Preheat(preheatingImage *PreheatImage) (*PreheatingStatu
 		},
 	}
 	events = append(events, event)
-	var payload = cm.Notification{
+	var payload = notification.Notification{
 		Events: events,
 	}
 	_, err := client.GetHTTPClient(kd.instance.Insecure).Post(url, kd.getCred(), payload, nil)
diff --git a/src/pkg/p2p/preheat/provider/mock.go b/src/pkg/p2p/preheat/provider/mock.go
index d72938932..31c46c1e2 100644
--- a/src/pkg/p2p/preheat/provider/mock.go
+++ b/src/pkg/p2p/preheat/provider/mock.go
@@ -16,13 +16,12 @@ package provider
 
 import (
 	"encoding/json"
+	"github.com/goharbor/harbor/src/pkg/p2p/preheat/models/notification"
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
 	"strings"
 	"time"
-
-	cm "github.com/goharbor/harbor/src/common/models"
 )
 
 // This is a package to provide mock utilities.
@@ -111,8 +110,8 @@ func MockKrakenProvider() *httptest.Server {
 				return
 			}
 
-			var payload = &cm.Notification{
-				Events: []cm.Event{},
+			var payload = &notification.Notification{
+				Events: []notification.Event{},
 			}
 
 			if err := json.Unmarshal(data, payload); err != nil {