Fix to migrate not applying updates (only two tests fail now)