flax

27e94552 - Merge pull request #839 from jheek:fix-normalization-test-2