Forgot to import cPickle in the test script.