Fix migration state.

Change-Id: I6592455c906f55c5c78c8d03fd614285c733935f