Make workflow pickleable