diff --git a/pyproject.toml b/pyproject.toml
index 2a775e5..3694548 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -97,3 +97,6 @@ force_grid_wrap = 2
 [tool.flake8]
 max-line-length = 120
 extend-ignore = "E203"
+
+#[tool.pytest.ini_options]
+#addopts = "-p no:faulthandler"
diff --git a/tests/test_e2e.py b/tests/test_e2e.py
index 88e4393..433148a 100644
--- a/tests/test_e2e.py
+++ b/tests/test_e2e.py
@@ -1,5 +1,6 @@
 import pytest
 import xgboost as xgb
+from matplotlib import use
 from neptune.new.integrations.xgboost import NeptuneCallback
 
 try:
@@ -12,6 +13,8 @@
 
 @pytest.mark.parametrize("log_tree", [None, [0, 1, 2, 3]])
 def test_e2e(dataset, log_tree):
+    use("Agg")
+
     # Start a run
     run = init_run()