Remove TA requirement for the chest and replace with class diversity requirement for the chest. If you have more than 2 of any class, you don't get the chest. Quick and easy fix, although not ideal. Would make class stacking unfavorable though and people might actually try to learn to do fights with melee.
I also agree with that we shouldn't be able to swap classes inside the dungeon.