diff --git a/test/components/generators/test_openai_dalle.py b/test/components/generators/test_openai_dalle.py
index 95c8ee87ed..6c28e23dfb 100644
--- a/test/components/generators/test_openai_dalle.py
+++ b/test/components/generators/test_openai_dalle.py
@@ -10,7 +10,7 @@ def test_to_dict(self) -> None:
         generator = DALLEImageGenerator()
         data = generator.to_dict()
         assert data == {
-            "type": "dc_custom_component.components.generators.image_generator.DALLEImageGenerator",
+            "type": "haystack.components.generators.image_generator.DALLEImageGenerator",
             "init_parameters": {
                 "model": "dall-e-3",
                 "quality": "standard",
@@ -24,7 +24,7 @@ def test_to_dict(self) -> None:
 
     def test_from_dict(self) -> None:
         data = {
-            "type": "dc_custom_component.components.generators.image_generator.DALLEImageGenerator",
+            "type": "haystack.components.generators.image_generator.DALLEImageGenerator",
             "init_parameters": {
                 "model": "dall-e-3",
                 "quality": "standard",