WIP: Minor tweak to brain masking