Changed the tests according to the change.