diff --git a/tests/app/notifications/rest/test_callbacks.py b/tests/app/notifications/rest/test_callbacks.py
index eac7aa477..1d8cc85ca 100644
--- a/tests/app/notifications/rest/test_callbacks.py
+++ b/tests/app/notifications/rest/test_callbacks.py
@@ -3,7 +3,7 @@ import uuid
 from datetime import datetime
 from flask import json
 from freezegun import freeze_time
-from mock import call
+from unittest.mock import call
 
 import app.celery.tasks
 from app.dao.notifications_dao import (
diff --git a/tests/app/public_contracts/schemas/GET_notifications_return.json b/tests/app/public_contracts/schemas/GET_notifications_return.json
index 7939dd8f1..709cca9fe 100644
--- a/tests/app/public_contracts/schemas/GET_notifications_return.json
+++ b/tests/app/public_contracts/schemas/GET_notifications_return.json
@@ -14,6 +14,17 @@
     },
     "links": {
       "type": "object",
+      "properties" : {
+        "prev" : {
+          "type" : "string"
+        },
+        "next" : {
+          "type" : "string"
+        },
+        "last": {
+          "type" : "string"
+        }
+      },
       "additionalProperties": false
     },
     "page_size": {"type": "number"},