Thanks for catching this. I just updated it so that it should work with the latest flax.