Prod vs Staging is only for Enterprise. Added the enterprise only class.