diff --git a/source/rmg/main.py b/source/rmg/main.py index c95d25a0ee..d4eb1af389 100755 --- a/source/rmg/main.py +++ b/source/rmg/main.py @@ -101,7 +101,7 @@ def execute(inputFile, options): logging.info('Loading previous restart file...') f = open(os.path.join(settings.outputDirectory,'restart.pkl'), 'rb') species.speciesList = cPickle.load(f) - species.speciesCounter = len(species.speciesList) + species.speciesCounter = cPickle.load(f) reaction.reactionList = cPickle.load(f) reactionModel = cPickle.load(f) reactionSystems = cPickle.load(f) @@ -151,6 +151,7 @@ def execute(inputFile, options): logging.info('Saving restart file...') f = open(os.path.join(settings.outputDirectory,'restart.pkl'), 'wb') cPickle.dump(species.speciesList, f) + cPickle.dump(species.speciesCounter, f) cPickle.dump(reaction.reactionList, f) cPickle.dump(reactionModel, f) cPickle.dump(reactionSystems, f)