Move upgrade to ee section