Fix misplaced test class.