Also add capabilities on upgrade