diff --git a/.github/actions/e2e/action.yml b/.github/actions/e2e/action.yml
index 9a701f7..9fd3027 100644
--- a/.github/actions/e2e/action.yml
+++ b/.github/actions/e2e/action.yml
@@ -21,11 +21,6 @@ runs:
       shell: bash
       env:
         MPLLOCALFREETYPE: "1"
-    # - name: Setup Graphviz
-    #   uses: ts-graphviz/setup-graphviz@v1
-    #   with:
-    #     ubuntu-skip-apt-update: true
-    #     macos-skip-brew-update: true
 
     - name: Run tests
       working-directory: ${{ inputs.working_directory }}
diff --git a/tests/test_e2e.py b/tests/test_e2e.py
index d17c00c..7d4a8e4 100644
--- a/tests/test_e2e.py
+++ b/tests/test_e2e.py
@@ -6,6 +6,7 @@
 except ImportError:
     from neptune.new import Run, init_run
 
+import matplotlib as mpl
 import pytest
 from sklearn.cluster import KMeans
 from sklearn.dummy import (
@@ -16,6 +17,8 @@
 
 import neptune_sklearn as npt_utils
 
+mpl.use("agg")
+
 
 def test_classifier_summary(iris):
     with init_run() as run: