diff --git a/src/Entity/MigrationGroup.php b/src/Entity/MigrationGroup.php
index 579ae6789fd3749eba0d6d9f22e4ddb3986ea012..589f1d7d8ab6a11acb1aabc7fb3e1f1c8916f488 100644
--- a/src/Entity/MigrationGroup.php
+++ b/src/Entity/MigrationGroup.php
@@ -8,7 +8,7 @@
 namespace Drupal\migrate_plus\Entity;
 
 use Drupal\Core\Config\Entity\ConfigEntityBase;
-use Drupal\migrate\Entity\MigrationInterface;
+use Drupal\migrate\Plugin\MigrationInterface;
 
 /**
  * Defines the Migration Group entity.
diff --git a/src/Event/MigratePrepareRowEvent.php b/src/Event/MigratePrepareRowEvent.php
index edd10bcface8a5a196a368b76d41bf8aa8c4137d..e64ac4064fb8c35a114784c35b07029f0a67e52b 100644
--- a/src/Event/MigratePrepareRowEvent.php
+++ b/src/Event/MigratePrepareRowEvent.php
@@ -7,7 +7,7 @@
 
 namespace Drupal\migrate_plus\Event;
 
-use Drupal\migrate\Entity\MigrationInterface;
+use Drupal\migrate\Plugin\MigrationInterface;
 use Drupal\migrate\Plugin\MigrateSourceInterface;
 use Drupal\migrate\Row;
 use Symfony\Component\EventDispatcher\Event;
@@ -25,16 +25,16 @@ class MigratePrepareRowEvent extends Event {
   protected $row;
 
   /**
-   * Migration entity.
+   * Migration source plugin.
    *
    * @var \Drupal\migrate\Plugin\MigrateSourceInterface
    */
   protected $source;
 
   /**
-   * Migration entity.
+   * Migration plugin.
    *
-   * @var \Drupal\migrate\Entity\MigrationInterface
+   * @var \Drupal\migrate\Plugin\MigrationInterface
    */
   protected $migration;
 
@@ -47,7 +47,7 @@ class MigratePrepareRowEvent extends Event {
    * @param \Drupal\migrate\Plugin\MigrateSourceInterface $source
    *   Source plugin that is the source of the event.
    *
-   * @param \Drupal\migrate\Entity\MigrationInterface $migration
+   * @param \Drupal\migrate\Plugin\MigrationInterface $migration
    *   Migration entity.
    */
   public function __construct(Row $row, MigrateSourceInterface $source, MigrationInterface $migration) {
@@ -77,9 +77,9 @@ class MigratePrepareRowEvent extends Event {
   }
 
   /**
-   * Gets the migration entity.
+   * Gets the migration plugin.
    *
-   * @return \Drupal\migrate\Entity\MigrationInterface
+   * @return \Drupal\migrate\Plugin\MigrationInterface
    *   The migration entity being imported.
    */
   public function getMigration() {