flax

875b40fe - Merge pull request #3153 from levskaya:trafo_setup_fix