Push upgradeability into parent class