I think that's not a horrible idea. It's a way of keeping track of wand-y exploration and effort. It would give the staff yet another thing to keep track of, as all mage characters will now have that, but maybe it's not so bad?