diff --git a/batchflow/models/torch/base_mixins.py b/batchflow/models/torch/base_mixins.py
index 98a791da5..54252f0d7 100644
--- a/batchflow/models/torch/base_mixins.py
+++ b/batchflow/models/torch/base_mixins.py
@@ -6,10 +6,7 @@
 import torch
 
 from ...decorators import deprecated
-
-from ...utils_import import try_import
-plot = try_import(module='...plotter', package=__name__, attribute='plot',
-                  help='Try `pip install batchflow[image]`!')
+from ... import plot
 
 # Also imports `tensorboard`, if necessary