I'm not a big fan of pure stat stick weapons. I feel like staves could provide unique passive bonuses (like bonuses to spells or particular classes of spells in more interesting ways than just +ability points) for mages without being made useless as actual weapons. Wands or wand + shield pretty much outclass staves for everyone but a melee-staff build (which, however, is very good), and certain unique bonuses applying only to staves would help give it a bigger role as a standard caster weapon.

Slightly related, staff of magus should inherit on-hit effects of the staff. In fact, it'd be great if every skill inherited some % of the chance to apply on-hit effects of weapons. Throwing Knife might inherit 50-100% chance of the % to proc on the weapon, where whirlwind might inherit 25-33% of the chance. Those on-hit effects often felt like a waste (except blatantly OP stuff like 35% chance to stun for 2 turns on hit) later in the game in D:OS EE when you were primarily using skills instead of regular attacks, though this depended on the build somewhat. At the very least, some kind of talent to let you inherit 33-50% of chance of weapon procs on to your skills would be cool.