correction in plugins

dream/plugins/AddBatchStations.py

@@ -21,7 +21,8 @@ class AddBatchStations(plugin.InputPreparationPlugin):
         standardBatchUnits=0
         for node_id, node in nodes.iteritems():
             if node['_class']=='Dream.BatchSource':
-                standardBatchUnits=int(node['numberOfUnits'])    
+                print node
+                standardBatchUnits=int(node['batchNumberOfUnits'])    
         
         # loop in BatchScrapMachines to change the classes if need be       
         for node_id, node in nodes.iteritems():

dream/plugins/BatchesWIPSpreadsheet.py

@@ -22,7 +22,7 @@ class BatchesWIPSpreadsheet(plugin.InputPreparationPlugin):
     standardBatchUnits=0
     for node_id, node in nodes.iteritems():
         if node['_class']=='Dream.BatchSource':
-            standardBatchUnits=int(node['numberOfUnits']) 
+            standardBatchUnits=int(node['batchNumberOfUnits']) 
         node['wip']=[]
 
     if wipData: