Hook update for new Migration classes in 2.x branch.