Fix function pushing on decorators to be mpi/ipython agnostic.