Fix 2.0.1 migration error.