Fixed a pickled args mistake