That was my understanding as well. Nico didn't take the Witchbreaker's staff (because that would totally fuck up the time stream), her torture by her ancestor simply made Nico and her staff more powerful. The Staff of One evolved and that evolution included a dragon.