diff --git a/tests/openvino/test_diffusion.py b/tests/openvino/test_diffusion.py
index d8e8e76030..03c1e20486 100644
--- a/tests/openvino/test_diffusion.py
+++ b/tests/openvino/test_diffusion.py
@@ -248,7 +248,7 @@ def test_compare_to_diffusers(self, model_arch: str):
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
 
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     def test_image_reproducibility(self, model_arch: str):
@@ -406,7 +406,7 @@ def test_compare_to_diffusers(self, model_arch: str):
             # Compare model outputs
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     def test_image_reproducibility(self, model_arch: str):
@@ -536,7 +536,7 @@ def test_compare_to_diffusers(self, model_arch: str):
             # Compare model outputs
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     @pytest.mark.run_slow
@@ -602,4 +602,4 @@ def test_safety_checker(self, model_arch: str):
             # Compare model outputs
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)