diff --git a/pymc_marketing/mmm/base.py b/pymc_marketing/mmm/base.py
index 28d1a45d..98d1e694 100644
--- a/pymc_marketing/mmm/base.py
+++ b/pymc_marketing/mmm/base.py
@@ -7,6 +7,7 @@
     ismethoddescriptor,
 )
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+import types
 
 import arviz as az
 import matplotlib.pyplot as plt
diff --git a/pymc_marketing/mmm/delayed_saturated_mmm.py b/pymc_marketing/mmm/delayed_saturated_mmm.py
index 964b1f31..df1450b9 100644
--- a/pymc_marketing/mmm/delayed_saturated_mmm.py
+++ b/pymc_marketing/mmm/delayed_saturated_mmm.py
@@ -1,6 +1,5 @@
 import json
 from pathlib import Path
-import types
 from typing import Any, Dict, List, Optional, Union
 
 import arviz as az