add cleanup after upgrade