Fix broken tests according to yangtools changes