diff --git a/core/includes/update.inc b/core/includes/update.inc
index a9b5de9..eaa1fc8 100644
--- a/core/includes/update.inc
+++ b/core/includes/update.inc
@@ -210,7 +210,7 @@ function update_do_one($module, $number, $dependency_map, &$context) {
   }
 
   // Record the schema update if it was completed successfully.
-  if ($context['finished'] == 1 && empty($ret['#abort'])) {
+  if ($context['finished'] >= 1 && empty($ret['#abort'])) {
     drupal_set_installed_schema_version($module, $number);
   }