Upgrade migration script to the latest model.

Change-Id: I6d463794e4f2764f27ea1dcbeee7755019d44bc1
Merge "Upgrade migration script to the latest model."