diff --git a/qim3d/processing/_local_thickness.py b/qim3d/processing/_local_thickness.py
index 60fbc516b95035468623367fe8c55a74ea693054..033b8898495d0c8798af2859781a9da82da444d7 100644
--- a/qim3d/processing/_local_thickness.py
+++ b/qim3d/processing/_local_thickness.py
@@ -4,7 +4,7 @@ import numpy as np
 from typing import Optional
 from qim3d.utils._logger import log
 import qim3d
-
+from IPython.display import display
 
 def local_thickness(
     image: np.ndarray,
diff --git a/qim3d/processing/_structure_tensor.py b/qim3d/processing/_structure_tensor.py
index 0a72815352c1307304d267966a1c1f4611ac3aed..d2b6e76ce628dbb127cdc26d073a1ac2597cf49d 100644
--- a/qim3d/processing/_structure_tensor.py
+++ b/qim3d/processing/_structure_tensor.py
@@ -4,7 +4,7 @@ from typing import Tuple
 import logging
 import numpy as np
 from qim3d.utils._logger import log
-
+from IPython.display import display
 
 def structure_tensor(
     vol: np.ndarray,