diff --git a/yii/swiftmailer/Mailer.php b/yii/swiftmailer/Mailer.php
index 2f8bbe7..1d5bc5c 100644
--- a/yii/swiftmailer/Mailer.php
+++ b/yii/swiftmailer/Mailer.php
@@ -8,7 +8,7 @@
 namespace yii\swiftmailer;
 
 use yii\base\InvalidConfigException;
-use yii\email\BaseMailer;
+use yii\mail\BaseMailer;
 
 /**
  * Mailer based on SwiftMailer library.
diff --git a/yii/swiftmailer/Message.php b/yii/swiftmailer/Message.php
index e850208..7cf3032 100644
--- a/yii/swiftmailer/Message.php
+++ b/yii/swiftmailer/Message.php
@@ -7,7 +7,7 @@
 
 namespace yii\swiftmailer;
 
-use yii\email\BaseMessage;
+use yii\mail\BaseMessage;
 
 /**
  * Email message based on SwiftMailer library.