diff --git a/dummy-service/dummyService.py b/dummy-service/dummyService.py
index 7459b537f608ccc005e8c5dd2330b1210d04f5a3..842baec2fd56a27d60f2a6020448a923e1902980 100644
--- a/dummy-service/dummyService.py
+++ b/dummy-service/dummyService.py
@@ -2,6 +2,7 @@ import logging
 from typing import List
 
 from interface.TextMiningService import TextMiningService, Publication
+from models.coocurrence import CoOccurrence
 
 logger = logging.getLogger(__name__)
 
@@ -13,7 +14,7 @@ class DummyService(TextMiningService):
     def get_mentions(self, entities: List, limit: int = 20) -> List[Publication]:
         return [Publication(pm_id="00000" + str(i)) for i in range(20)]
 
-    def get_co_occurrences(self, entity: str) -> List[str]:
+    def get_co_occurrences(self, entity: str, limit: int = 20) -> List[CoOccurrence]:
         pass
 
 
diff --git a/jensenlab-service/jensenLabService.py b/jensenlab-service/jensenLabService.py
index c2b9b98ea0e1a608783b16dd1cee2b4fe38789a9..a3a77e00ec099c2ee246bc31897b3ee3640d4a9d 100644
--- a/jensenlab-service/jensenLabService.py
+++ b/jensenlab-service/jensenLabService.py
@@ -29,7 +29,7 @@ class JensenLabService(TextMiningService):
         publications_ids_intersection = set.intersection(*publications_ids)
         return [Publication(pm_id=pid) for pid in publications_ids_intersection]
 
-    def get_co_occurrences(self, entity: str) -> List[str]:
+    def get_co_occurrences(self, entity: str, limit: int = 20) -> List[str]:
         pass
 
     @staticmethod