fixed a bug removing a class