diff --git a/src/Plugin/migrate/source/Agreement.php b/src/Plugin/migrate/source/Agreement.php
index da1c181..8c6fdca 100644
--- a/src/Plugin/migrate/source/Agreement.php
+++ b/src/Plugin/migrate/source/Agreement.php
@@ -2,7 +2,7 @@
 
 namespace Drupal\agreement\Plugin\migrate\source;
 
-use Drupal\migrate\Plugin\migrate\source\SqlBase;
+use Drupal\migrate_drupal\Plugin\migrate\source\DrupalSqlBase;
 
 /**
  * Agreement migrate source plugin.
@@ -12,7 +12,7 @@ use Drupal\migrate\Plugin\migrate\source\SqlBase;
  *   source_module = "agreement",
  * )
  */
-class Agreement extends SqlBase {
+class Agreement extends DrupalSqlBase {
 
   /**
    * {@inheritdoc}
diff --git a/src/Plugin/migrate/source/d7/AgreementType.php b/src/Plugin/migrate/source/d7/AgreementType.php
index f153ae7..37ca8e3 100644
--- a/src/Plugin/migrate/source/d7/AgreementType.php
+++ b/src/Plugin/migrate/source/d7/AgreementType.php
@@ -2,7 +2,7 @@
 
 namespace Drupal\agreement\Plugin\migrate\source\d7;
 
-use Drupal\migrate\Plugin\migrate\source\SqlBase;
+use Drupal\migrate_drupal\Plugin\migrate\source\DrupalSqlBase;
 
 /**
  * Agreement type migrate source plugin.
@@ -12,7 +12,7 @@ use Drupal\migrate\Plugin\migrate\source\SqlBase;
  *   source_module = "agreement"
  * )
  */
-class AgreementType extends SqlBase {
+class AgreementType extends DrupalSqlBase {
 
   /**
    * {@inheritdoc}