ENH: Updated after atlas changes