Fix pickling support