diff --git a/mdpath/tests/test_visualization.py b/mdpath/tests/test_visualization.py
index 0920ef5..ca631d4 100644
--- a/mdpath/tests/test_visualization.py
+++ b/mdpath/tests/test_visualization.py
@@ -1,3 +1,6 @@
+import matplotlib
+matplotlib.use("Agg")
+
 import pytest
 import tempfile
 import os