Image for training freeze after first training

[remybonnav]

Using a multichannel image for training.

After adding some traces to different classes and doing training, the image freeze and I can't add more traces or zoom in/out. I need to click somewhere outside imageJ to unfreeze the image