`InstanceGeneratorBatch` does not save the whole object

InstanceGeneratorBatch only saves the dictionary of instances, and not the object itself. As a result, other properties of the object is lost on save.