Put the class in the right place