Minor bugfix for pickled code.