Oups, forgot to add the new classes