flax
fix deprecation warning
#3981
Merged

Loading