Add upgrade to enable collection behavior.