flax

68caa8a2 - Merge pull request #1595 from dan-zheng:fix-typos