Move the context_switch flag into the Capability