diff --git a/interface/TextMiningService.py b/interface/TextMiningService.py
index bd0c219838daaff0ad764e65ed96c9cb3a3d90db..48f38bec8670d06ca1c335b829298471d8cf7518 100644
--- a/interface/TextMiningService.py
+++ b/interface/TextMiningService.py
@@ -1,15 +1,7 @@
 from abc import ABCMeta, abstractmethod
 from typing import List
-
-
-class Publication:
-    def __init__(self, pmc_id=None, pm_id=None, doi=None, preprint_id=None, other_id=None):
-        self.pmc_id = pmc_id
-        self.pm_id = pm_id
-        self.doi = doi
-        self.preprint_id = preprint_id
-        self.other_id = other_id
-
+from models.publication import Publication
+from models.coocurrence import CoOccurrence
 
 class TextMiningService(metaclass=ABCMeta):