Fix feature test

- Fix feature test by adding class to school.