Update test output to include a class