diff --git a/src/MigrateTrait.php b/src/MigrateTrait.php
index ab1bc951..f23d6526 100644
--- a/src/MigrateTrait.php
+++ b/src/MigrateTrait.php
@@ -9,6 +9,18 @@
  */
 trait MigrateTrait {
 
+  /**
+   * Gets the limit.
+   *
+   * @return int
+   *   The limit.
+   */
+  public function getLimit() : int {
+    $limit = getenv('MIGRATE_LIMIT') ?: 0;
+
+    return (int) $limit;
+  }
+
   /**
    * Checks if we're doing partial migrate.
    *