diff --git a/src/remove_ai_watermarks/noai/ctrlregen/engine.py b/src/remove_ai_watermarks/noai/ctrlregen/engine.py
index be12225..abed690 100644
--- a/src/remove_ai_watermarks/noai/ctrlregen/engine.py
+++ b/src/remove_ai_watermarks/noai/ctrlregen/engine.py
@@ -36,9 +36,10 @@ _HAS_COLOR_MATCHER = False
 _HAS_DIFFUSERS = False
 
 try:
-    from remove_ai_watermarks.noai.ctrlregen.pipeline import CustomCtrlRegenPipeline
     from diffusers import AutoencoderKL, ControlNetModel, UniPCMultistepScheduler
 
+    from remove_ai_watermarks.noai.ctrlregen.pipeline import CustomCtrlRegenPipeline
+
     _HAS_DIFFUSERS = True
 except ImportError:
     AutoencoderKL = None  # type: ignore[assignment,misc]