Test should adapt to the class changes.