diff --git a/packages/pipeline/src/pyearthtools/pipeline/controller.py b/packages/pipeline/src/pyearthtools/pipeline/controller.py
index 568f3b9c..ad73dc2b 100644
--- a/packages/pipeline/src/pyearthtools/pipeline/controller.py
+++ b/packages/pipeline/src/pyearthtools/pipeline/controller.py
@@ -773,7 +773,7 @@ def save(self, path: Optional[Union[str, Path]] = None, only_steps: bool = False
 
     def _ipython_display_(self):
         """Override for repr of `Pipeline`, shows initialisation arguments and graph"""
-        from IPython.core.display import display, HTML
+        from IPython.display import display, HTML
 
         display(HTML(self._repr_html_()))