diff --git a/monstereo/run.py b/monstereo/run.py
index b7c196c..0f00fac 100644
--- a/monstereo/run.py
+++ b/monstereo/run.py
@@ -5,7 +5,6 @@ try:
     from openpifpaf.network.nets import cli as openpifpaf_cli
 except ImportError:
     from openpifpaf.network.factory import cli as openpifpaf_cli
-from openpifpaf.network import nets
 from openpifpaf import decoder
 
 
diff --git a/monstereo/train/trainer.py b/monstereo/train/trainer.py
index 2990921..e7dd6e7 100644
--- a/monstereo/train/trainer.py
+++ b/monstereo/train/trainer.py
@@ -59,7 +59,7 @@ class Trainer:
         self.lr = lr
         self.sched_step = sched_step
         self.sched_gamma = sched_gamma
-        self.clusters = ['10', '20', '30', '50', '>50']
+        self.clusters = ['10', '20', '30', '40']
         self.hidden_size = hidden_size
         self.n_stage = n_stage
         self.dir_out = dir_out
diff --git a/monstereo/utils/misc.py b/monstereo/utils/misc.py
index c3ab6eb..fd198ab 100644
--- a/monstereo/utils/misc.py
+++ b/monstereo/utils/misc.py
@@ -20,14 +20,14 @@ def append_cluster(dic_jo, phase, xx, ys, kps):
         dic_jo[phase]['clst']['30']['kps'].append(kps)
         dic_jo[phase]['clst']['30']['X'].append(xx)
         dic_jo[phase]['clst']['30']['Y'].append(ys)
-    elif ys[3] < 50:
-        dic_jo[phase]['clst']['50']['kps'].append(kps)
-        dic_jo[phase]['clst']['50']['X'].append(xx)
-        dic_jo[phase]['clst']['50']['Y'].append(ys)
+    elif ys[3] <= 40:
+        dic_jo[phase]['clst']['40']['kps'].append(kps)
+        dic_jo[phase]['clst']['40']['X'].append(xx)
+        dic_jo[phase]['clst']['40']['Y'].append(ys)
     else:
-        dic_jo[phase]['clst']['>50']['kps'].append(kps)
-        dic_jo[phase]['clst']['>50']['X'].append(xx)
-        dic_jo[phase]['clst']['>50']['Y'].append(ys)
+        dic_jo[phase]['clst']['>40']['kps'].append(kps)
+        dic_jo[phase]['clst']['>40']['X'].append(xx)
+        dic_jo[phase]['clst']['>40']['Y'].append(ys)
 
 
 def get_task_error(dd):