Fix Class Edit