diff --git a/tupak/core/utils.py b/tupak/core/utils.py
index 711e50383da36a9f99787001dce1e65612be6745..e31176ed1e708d5b135d2bce6cf685f1a808a3bb 100644
--- a/tupak/core/utils.py
+++ b/tupak/core/utils.py
@@ -452,12 +452,13 @@ else:
     logger.debug('No $DISPLAY environment variable found, so importing \
                    matplotlib.pyplot with non-interactive "Agg" backend.')
     import matplotlib
+    import matplotlib.pylot as plt
     non_gui_backends = matplotlib.rcsetup.non_interactive_bk
     for backend in non_gui_backends:
         try:
             logger.debug("Trying backend {}".format(backend))
             matplotlib.use(backend, warn=False)
-            matplotlib.pyplot.switch_backend(backend)
+            plt.switch_backend(backend)
             break
         except Exception as e:
             print(traceback.format_exc())