diff --git a/tests/framework/console/controllers/DbMessageControllerTest.php b/tests/framework/console/controllers/DbMessageControllerTest.php
index 8780f68..2869538 100644
--- a/tests/framework/console/controllers/DbMessageControllerTest.php
+++ b/tests/framework/console/controllers/DbMessageControllerTest.php
@@ -73,7 +73,6 @@ class DbMessageControllerTest extends BaseMessageControllerTest
     }
     
     /**
-     * @throws \yii\base\InvalidParamException
      * @throws \yii\db\Exception
      * @throws \yii\base\InvalidConfigException
      * @return \yii\db\Connection
diff --git a/tests/framework/db/CommandTest.php b/tests/framework/db/CommandTest.php
index e7e4cb5..dbf7047 100644
--- a/tests/framework/db/CommandTest.php
+++ b/tests/framework/db/CommandTest.php
@@ -433,7 +433,7 @@ SQL;
      * Test INSERT INTO ... SELECT SQL statement with wrong query object
      *
      * @dataProvider invalidSelectColumns
-     * @expectedException \yii\base\InvalidParamException
+     * @expectedException \yii\base\InvalidArgumentException
      * @expectedExceptionMessage Expected select query object with enumerated (named) parameters
      */
     public function testInsertSelectFailed($invalidSelectColumns)