diff --git a/phi/vis/_dash/colormaps.py b/phi/vis/_dash/colormaps.py
index c4d9ffcd4..54d1ab782 100644
--- a/phi/vis/_dash/colormaps.py
+++ b/phi/vis/_dash/colormaps.py
@@ -4,8 +4,6 @@
 
 import numpy as np
 
-from phi.math.backend import ML_LOGGER
-
 ORANGE_WHITE_BLUE = [
     [0.      ,  22.00000005,   1.00000035,  76.0000011 ],
     [0.030334,  28.999875  ,   5.999997  , 114.9999    ],
diff --git a/phi/vis/_dash/dash_gui.py b/phi/vis/_dash/dash_gui.py
index a72deb3d3..94e568a65 100644
--- a/phi/vis/_dash/dash_gui.py
+++ b/phi/vis/_dash/dash_gui.py
@@ -14,7 +14,6 @@
 from .viewer import build_viewers, REFRESH_INTERVAL
 from .player_controls import build_status_bar, build_player_controls, PLAYING, STEP_COMPLETE
 from .._vis_base import Gui, VisModel
-from ...math.backend import ML_LOGGER
 
 
 class DashGui(Gui):
diff --git a/phi/vis/_io.py b/phi/vis/_io.py
index 2ca6af01c..196e7d42b 100644
--- a/phi/vis/_io.py
+++ b/phi/vis/_io.py
@@ -6,7 +6,7 @@
 from phi import math
 from phi.field import Scene
 from phi.math import shape, wrap, channel, spatial, batch
-from phi.math.backend import ML_LOGGER
+from phiml.backend import ML_LOGGER
 
 
 @math.broadcast