I found that storing a Creature with a brain augment will not drop the augment. It just vanishes.
It isn't a major bug, just a bit annoying if you have to recraft the augment because you forget to take it out (and it is a tiny resource drain if you are as forgetful as me).