Fix in the upgrade steps