diff --git a/tests/openvino/test_diffusion.py b/tests/openvino/test_diffusion.py
index 6eec57694a..0ead50c2e8 100644
--- a/tests/openvino/test_diffusion.py
+++ b/tests/openvino/test_diffusion.py
@@ -110,7 +110,7 @@ def test_ov_pipeline_class_dispatch(self, model_arch: str):
 
         self.assertEqual(ov_pipeline.auto_model_class, auto_pipeline.__class__)
 
-        auto_pipeline = auto_pipeline.from_pretrained(MODEL_NAMES[model_arch])
+        auto_pipeline = DiffusionPipeline.from_pretrained(MODEL_NAMES[model_arch])
         ov_pipeline = OVDiffusionPipeline.from_pretrained(MODEL_NAMES[model_arch])
 
         self.assertEqual(ov_pipeline.auto_model_class, auto_pipeline.__class__)