flax

54634f7e - Merge pull request #4120 from dcharatan:parent_type_fix