Now using the correct document class in the test