diff --git a/pymc_marketing/mmm/base.py b/pymc_marketing/mmm/base.py
index 98d1e694..b4117cf9 100644
--- a/pymc_marketing/mmm/base.py
+++ b/pymc_marketing/mmm/base.py
@@ -15,6 +15,7 @@
 import numpy as np
 import pandas as pd
 import pymc as pm
+from pymc import str_for_dist
 import seaborn as sns
 from pytensor.tensor import TensorVariable
 from sklearn.pipeline import Pipeline