diff --git a/pyproject.toml b/pyproject.toml
index 67c8cd6ea38d1c4eb6319270a538b5ba35b5eacf..434116f2709ad4c12b6fc2ae0e9851d6650467e6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "MaggotUBA-adapter"
-version = "0.16"
+version = "0.16.1"
 description = "Interface between MaggotUBA and the Nyx tagging UI"
 authors = ["François Laurent"]
 license = "MIT"
@@ -14,7 +14,7 @@ maggotuba-core = {git = "https://gitlab.pasteur.fr/nyx/MaggotUBA-core", tag = "v
 torch = "^1.11.0"
 numpy = "^1.19.3"
 protobuf = "3.9.2"
-taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.14.1"}
+taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15"}
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
diff --git a/src/maggotuba/data/make_dataset.py b/src/maggotuba/data/make_dataset.py
index 8595288f58dbe4a2619f5922237632488b902978..b193efc21a784515113a011205e803f297a3b7d7 100644
--- a/src/maggotuba/data/make_dataset.py
+++ b/src/maggotuba/data/make_dataset.py
@@ -59,10 +59,10 @@ def make_dataset(backend, labels_expected=False, trxmat_only=False,
 
             print("generating a larva_dataset file...")
             # generate a larva_dataset_*.hdf5 file in data/interim/{instance}/
-            balance = isinstance(balancing_strategy, str) and balancing_strategy.lower() == 'maggotuba'
             out = backend.generate_dataset(backend.raw_data_dir(),
                                            past_future_extensions=False,
-                                           balance=balance, **kwargs)
+                                           balancing_strategy=balancing_strategy,
+                                           **kwargs)
             print(f"larva_dataset file generated: {out}")