fix what looks to be a copy-paste error or typo, caused pickle error

I ran across this issue while working on a classification script. Seems simple enough but I don't have much background with the code-base. What do you think?