diff --git a/modules/islandora_core_feature/islandora_core_feature.install b/modules/islandora_core_feature/islandora_core_feature.install
index 5c1292b2..eabd5672 100644
--- a/modules/islandora_core_feature/islandora_core_feature.install
+++ b/modules/islandora_core_feature/islandora_core_feature.install
@@ -125,7 +125,7 @@ function islandora_core_feature_update_8001(&$sandbox) {
     // Clean up.
     foreach ($tables as $table) {
       $database->schema()->dropTable($table . '_o');
-      if ($sandbox['progress'] + $limit >= $progress[$table]['size']) {
+      if ($sandbox['progress'] + $limit >= $sandbox[$table]['size']) {
         $sandbox[$table]['done'] = TRUE;
       }
     }