diff --git a/core/modules/migrate/src/Entity/Migration.php b/core/modules/migrate/src/Entity/Migration.php
index 26a5520..a76d1e6 100644
--- a/core/modules/migrate/src/Entity/Migration.php
+++ b/core/modules/migrate/src/Entity/Migration.php
@@ -396,6 +396,7 @@ public function getProcess() {
    */
   public function setProcess(array $process) {
     $this->process = $process;
+    return $this;
   }
 
   /**
@@ -410,6 +411,7 @@ public function getSystemOfRecord() {
    */
   public function setSystemOfRecord($system_of_record) {
     $this->systemOfRecord = $system_of_record;
+    return $this;
   }
 
   /**
@@ -423,7 +425,8 @@ public function isTrackLastImported() {
    * {@inheritdoc}
    */
   public function setTrackLastImported($track_last_imported) {
-    return $this->trackLastImported = (bool) $track_last_imported;
+    $this->trackLastImported = (bool) $track_last_imported;
+    return $this;
   }
 
   /**
diff --git a/core/modules/migrate/src/Entity/MigrationInterface.php b/core/modules/migrate/src/Entity/MigrationInterface.php
index f18cc9e..44ba4f8 100644
--- a/core/modules/migrate/src/Entity/MigrationInterface.php
+++ b/core/modules/migrate/src/Entity/MigrationInterface.php
@@ -192,6 +192,8 @@ public function getProcess();
    *
    * @param array $process
    *   The configuration describing the process plugins.
+   *
+   * @return $this
    */
   public function setProcess(array $process);
 
@@ -208,6 +210,8 @@ public function getSystemOfRecord();
    *
    * @param string $system_of_record
    *   The system of record of the migration.
+   *
+   * @return $this
    */
   public function setSystemOfRecord($system_of_record);
 
@@ -224,6 +228,8 @@ public function isTrackLastImported();
    *
    * @param bool $track_last_imported
    *   Boolean value to indicate if the migration should track last import time.
+   *
+   * @return $this
    */
   public function setTrackLastImported($track_last_imported);