diff --git a/tests/CommandTest.php b/tests/CommandTest.php
index 84beb70fa..8744451d1 100644
--- a/tests/CommandTest.php
+++ b/tests/CommandTest.php
@@ -110,9 +110,9 @@ public function testBooleanValuesBatchInsert(): void
         $db = $this->getConnection(true);
 
         $command = $db->createCommand();
-        $command->insertBatch('{{bool_values}}', [[true], [false]], ['bool_col']);
+        $batchCommand = $command->insertBatch('{{bool_values}}', [[true], [false]], ['bool_col']);
 
-        $this->assertSame(2, $command->execute());
+        $this->assertSame(2, $batchCommand->execute());
 
         $command->setSql(
             <<<SQL