Index: pifr.install
===================================================================
RCS file: /cvs/drupal-contrib/contributions/modules/project_issue_file_review/pifr.install,v
retrieving revision 1.38
diff -u -r1.38 pifr.install
--- pifr.install	3 Nov 2009 16:22:22 -0000	1.38
+++ pifr.install	12 Nov 2009 19:02:33 -0000
@@ -315,7 +315,7 @@
 
   // Upon completion of the migration check the number of rows in the new
   // tables to ensure that all the data was migrated successfully.
-  if ($ret['#finished'] == 1) {
+  if ($ret['#finished'] >= 1) {
     $test_count = db_result(db_query('SELECT COUNT(test_id) FROM {pifr_test} WHERE type = %d', PIFR_SERVER_TEST_TYPE_FILE));
     $result_count = db_result(db_query('SELECT COUNT(result_id) FROM {pifr_result}'));
 
@@ -326,6 +326,9 @@
     $line_count = count(file($sandbox['map_file']));
     $success = ($line_count == $test_count);
     $ret[] = pifr_update_message('Generated map file [' . $sandbox['map_file'] . '].', $success);
+
+    // Ensure that the operation finishes.
+    $ret['#finished'] = 1;
   }
 
   return $ret;
@@ -461,11 +464,14 @@
 
   // Upon completion of the migration check the number of rows in the new
   // tables to ensure that all the data was migrated successfully.
-  if ($ret['#finished'] == 1) {
+  if ($ret['#finished'] >= 1) {
     $log_count = db_result(db_query('SELECT COUNT(log_id) FROM {pifr_log}'));
 
     $success = ($log_count == $sandbox['max']);
     $ret[] = pifr_update_message('Migrated [' . number_format($sandbox['max']) . '] log entries.', $success);
+
+    // Ensure that the operation finishes.
+    $ret['#finished'] = 1;
   }
 
   return $ret;