fix upgrade done exception
fix upgrade to stop after done
fix image for upgrades