diff --git a/shapash/backend/shap_backend.py b/shapash/backend/shap_backend.py
index ec6867af..4771758c 100644
--- a/shapash/backend/shap_backend.py
+++ b/shapash/backend/shap_backend.py
@@ -26,7 +26,7 @@ def __init__(self, model, preprocessing=None, masker=None, explainer_args=None,
         else:
             if shap.explainers.Linear.supports_model_with_masker(model, self.masker):
                 self.explainer = shap.Explainer(model=model, masker=self.masker)
-            elif shap.explainers.Tree.supports_model_with_masker(model, self.masker):
+            elif shap.explainers.Tree.supports_model_with_masker(model, None):
                 self.explainer = shap.Explainer(model=model)
             elif shap.explainers.Additive.supports_model_with_masker(model, self.masker):
                 self.explainer = shap.Explainer(model=model, masker=self.masker)