Maybe we could call this `context_class` to make explicit it is a class. WDYT?