diff --git a/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py b/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py
index ba81083..e1304e0 100644
--- a/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py
+++ b/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py
@@ -11,7 +11,7 @@
 from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
 from packaging import version
 from diffusers.models import AutoencoderKL, UNet2DConditionModel
-from diffusers.pipeline_utils import DiffusionPipeline
+from diffusers import DiffusionPipeline
 from diffusers.schedulers import (
     DDIMScheduler,
     DPMSolverMultistepScheduler,
@@ -560,4 +560,4 @@ def __call__(
         if not return_dict:
             return (image, has_nsfw_concept)
 
-        return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)
\ No newline at end of file
+        return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)