fixed a class mistake