By examining neural responses from tree shrews performing hierarchical decision tasks with rule reversals, the authors identify a thalamocortical mechanism for regulating cognitive flexibility.