Made a mistake with the classes

It's fixed now