diff --git a/tests/test_analyst_data.py b/tests/test_analyst_data.py
index d594c0f..4672960 100644
--- a/tests/test_analyst_data.py
+++ b/tests/test_analyst_data.py
@@ -68,7 +68,8 @@ class TestAnalystData(unittest.TestCase):
         misp_object.add_attribute('filename', 'foo.exe')
         self._attach_analyst_data(misp_object.attributes[0])
         
-    def _attach_analyst_data(self, container) -> None:
+    def _attach_analyst_data(
+            self, container: MISPAttribute | MISPEvent | MISPEventReport | MISPObject) -> None:
         object_type = container._analyst_data_object_type
         note1 = container.add_note(note='note1')
         opinion1 = note1.add_opinion(opinion=25, comment='Disagree')