Fix pickle tests.
1 file changed