Revert changes to pickle test