Tweaking class test ready for new website