Adjust to yangtools-2.0.0 changes