I bought this Wiha x Xiaomi kit a few years ago.
What's cool about this kit is that you can purchase additional bits (micro bits, 4mm if I recall correctly) from Wiha's website. I've only replaced the pentalobe bit once compared to quite a bit when I had the iFixit kit. In a pinch, you can even use the iFixit bits in this driver. Beware though, it's a very thin driver. Not the most comfortable but I love it and swear by its bits.
But on the one you linked, yes. You do need to replace the driver. I apologize, I probably should have specified which driver I was using lol.