diff --git a/tests/CommandTest.php b/tests/CommandTest.php
index 387e4617..13a2d037 100644
--- a/tests/CommandTest.php
+++ b/tests/CommandTest.php
@@ -241,9 +241,10 @@ public function testUpdate(
         array $columns,
         array|string $conditions,
         array $params,
-        string $expected
+        array $expectedValues,
+        int $expectedCount,
     ): void {
-        parent::testUpdate($table, $columns, $conditions, $params, $expected);
+        parent::testUpdate($table, $columns, $conditions, $params, $expectedValues, $expectedCount);
     }
 
     /**
diff --git a/tests/QueryBuilderTest.php b/tests/QueryBuilderTest.php
index cf1a397b..43e9eb29 100644
--- a/tests/QueryBuilderTest.php
+++ b/tests/QueryBuilderTest.php
@@ -523,6 +523,18 @@ public function testSelectExists(string $sql, string $expected): void
         parent::testSelectExists($sql, $expected);
     }
 
+    /** @dataProvider \Yiisoft\Db\Mssql\Tests\Provider\QueryBuilderProvider::update */
+    public function testUpdate(
+        string $table,
+        array $columns,
+        array|string $condition,
+        array $params,
+        string $expectedSql,
+        array $expectedParams,
+    ): void {
+        parent::testUpdate($table, $columns, $condition, $params, $expectedSql, $expectedParams);
+    }
+
     /**
      * @dataProvider \Yiisoft\Db\Mssql\Tests\Provider\QueryBuilderProvider::upsert
      *