changed to the correct clas