fix unpickable object on store_output