diff --git a/tests/model_tests/diffusion/model_settings.py b/tests/model_tests/diffusion/model_settings.py
index 0a7571b7b..60e1e26bf 100644
--- a/tests/model_tests/diffusion/model_settings.py
+++ b/tests/model_tests/diffusion/model_settings.py
@@ -179,5 +179,6 @@ DIFFUSION_TEST_SETTINGS = {
         model="krea/Krea-2-Turbo",
         builder=diff_model_builders.tiny_krea2_builder,
         supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE],
+        model_type_marker=ModelTypeMarker.DIFFUSION,
     ),
 }