Fix: change upgrade process