diff --git a/qim3d/processing/structure_tensor_.py b/qim3d/processing/structure_tensor_.py
index 86d17213a825251563c333ddafd5e91f44c27ab4..97a0ba731bda4fae777cf00db4aa91ccfaeb7aa7 100644
--- a/qim3d/processing/structure_tensor_.py
+++ b/qim3d/processing/structure_tensor_.py
@@ -3,6 +3,7 @@
 from typing import Tuple
 import logging
 import numpy as np
+from qim3d.utils.logger import log
 
 
 def structure_tensor(
diff --git a/qim3d/viz/structure_tensor.py b/qim3d/viz/structure_tensor.py
index 2255e35b5f3fa0b8718136a851163d98d8fab985..e3dcbcb4f12a2eb46ac75089f15de3202132275e 100644
--- a/qim3d/viz/structure_tensor.py
+++ b/qim3d/viz/structure_tensor.py
@@ -4,6 +4,8 @@ import matplotlib.pyplot as plt
 from matplotlib.gridspec import GridSpec
 import ipywidgets as widgets
 import logging
+from qim3d.utils.logger import log
+
 
 previous_logging_level = logging.getLogger().getEffectiveLevel()
 logging.getLogger().setLevel(logging.CRITICAL)