I am repairing my mates Niteflux Vision Stick. The dirver seems stuffed. It is only giving 100mA to the LED on start up, then this drops to 3mA

Now, Is there a driver out there that uses a mom switch and can power a XR-E off one Li-Ion. At about 1000mA? maxFlex?? bFlex??

Or, does anyone know if I can replace the niteflux driver like for like??