fixed types in merge methods

2 files changed, 2 insertions, 2 deletions

diff --git a/sec_certs/certificate.py b/sec_certs/certificate.py
index ccf940b0..5229e193 100644
--- a/sec_certs/certificate.py
+++ b/sec_certs/certificate.py
@@ -35,7 +35,7 @@ class Certificate(ABC):
         raise NotImplementedError('Mot meant to be implemented')
 
     @abstractmethod
-    def merge(self, other):
+    def merge(self, other: 'Certificate'):
         raise NotImplementedError('Not meant to be implemented')
 
 
diff --git a/sec_certs/dataset.py b/sec_certs/dataset.py
index ccba554e..1f19ba23 100644
--- a/sec_certs/dataset.py
+++ b/sec_certs/dataset.py
@@ -83,7 +83,7 @@ class Dataset(ABC):
     def get_certs_from_web(self):
         pass
 
-    def merge_certs(self, certs: Dict[str, 'CommonCriteriaCert']):
+    def merge_certs(self, certs: Dict[str, 'Certificate']):
         """
         Merges dictionary of certificates into the dataset. Assuming they all are CommonCriteria certificates
         """