diff --git a/src/biolord/_model.py b/src/biolord/_model.py
index 21dd17a..c52e10b 100644
--- a/src/biolord/_model.py
+++ b/src/biolord/_model.py
@@ -837,7 +837,7 @@ def _compute_prediction(
         for attribute_ in target_attributes:
             categories_index = pd.Index(adata.obs[attribute_].values, dtype="category")
             classes_dataset[attribute_] = {}
-            for key_, _ in tqdm(zip(*np.unique(categories_index.values, return_counts=True), strict=True)):
+            for key_, _ in tqdm(zip(*np.unique(categories_index.values, return_counts=True))):
                 bool_category = categories_index.get_loc(key_)
 
                 adata_cur = adata[bool_category, :].copy()