Fixing bugs for Network MAML 3.2.2