diff --git a/tests/models/kosmos2_5/test_processor_kosmos2_5.py b/tests/models/kosmos2_5/test_processor_kosmos2_5.py
index ebefc0822..62ba5a10d 100644
--- a/tests/models/kosmos2_5/test_processor_kosmos2_5.py
+++ b/tests/models/kosmos2_5/test_processor_kosmos2_5.py
@@ -46,6 +46,8 @@ if is_vision_available():
 
 @require_vision
 class Kosmos2_5ProcessorTest(ProcessorTesterMixin, unittest.TestCase):
+    processor_class = Kosmos2_5Processor
+
     def setUp(self):
         self.tmpdirname = tempfile.mkdtemp()
         image_processor = Kosmos2_5ImageProcessor()