diff --git a/includes/base.inc b/includes/base.inc
index 05d19e8..0d7d87c 100644
--- a/includes/base.inc
+++ b/includes/base.inc
@@ -434,7 +434,7 @@ abstract class MigrationBase {
     }
 
     // Make sure we clear our semaphores in case of abrupt exit
-    register_shutdown_function(array($this, 'endProcess'));
+    drupal_register_shutdown_function(array($this, 'endProcess'));
   }
 
   /**