Add migration from 2.x