Using batch properly in drush scripts.

Minor fix to the batch implementation in drush scripts.