fixing notification on assignCurrentUpgrade + fixing upgrade issue in reset balances