Fixing bugs for Network MAML 3.2.2

Agree. Thanks for catching this Andre.