Fix py3 pickle err