Fix test class.