flax

149018f6 - Merge pull request #1023 from jheek:fix-module-attr-wrapping