diff --git a/pyproject.toml b/pyproject.toml
index 2f85c89f5168f58b7b8b16838dfda2a9718e03e4..13ddec87d5983a97e9a3e8da7970d9d8dfda0215 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "MaggotUBA-adapter"
-version = "0.16.2"
+version = "0.16.3"
 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.15"}
+taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15.2"}
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
diff --git a/src/maggotuba/models/modules.py b/src/maggotuba/models/modules.py
index 62ca0707bfc0f7d0b9ba8aa5b8721f2b54809565..f567e736e4054da91ba5aac825e8e82aa911519d 100644
--- a/src/maggotuba/models/modules.py
+++ b/src/maggotuba/models/modules.py
@@ -362,7 +362,7 @@ class MaggotClassifier(MaggotModule):
             if n_iterations is not None:
                 if isinstance(n_iterations, str):
                     n_iterations = map(int, n_iterations.split(','))
-                if isinstance(n_iterations, int):
+                if isinstance(n_iterations, (int, np.int64)):
                     n_pretraining_iter = n_iterations // 2
                     n_finetuning_iter = n_iterations // 2
                 else: