Fix test class docblock