diff --git a/stubs/ActiveQuery.phpstub b/stubs/ActiveQuery.phpstub
index 1f67f9b..f31a7c3 100644
--- a/stubs/ActiveQuery.phpstub
+++ b/stubs/ActiveQuery.phpstub
@@ -17,8 +17,8 @@ namespace yii\db;
  * @property bool $asArray
  * @property bool $multiple
  *
- * @method TModel|null one()
- * @method array<mixed, TModel> all()
+ * @method TModel|null one($db = null)
+ * @method array<mixed, TModel> all($db = null)
  *
  * @extends Query<TModel, TMultiple>
  */
diff --git a/stubs/ActiveQueryInterface.phpstub b/stubs/ActiveQueryInterface.phpstub
index f8f71db..e1b1138 100644
--- a/stubs/ActiveQueryInterface.phpstub
+++ b/stubs/ActiveQueryInterface.phpstub
@@ -13,9 +13,9 @@ namespace yii\db;
 /**
  * @psalm-template TModel
  *
- * @method TModel|null one()
- * @method array<mixed, TModel> all()
+ * @method TModel|null one($db = null)
+ * @method array<mixed, TModel> all($db = null)
  *
  * @extends QueryInterface<TModel>
  */
-class ActiveQueryInterface extends QueryInterface {}
\ No newline at end of file
+class ActiveQueryInterface extends QueryInterface {}
diff --git a/stubs/Query.phpstub b/stubs/Query.phpstub
index ed99b31..f3f2117 100644
--- a/stubs/Query.phpstub
+++ b/stubs/Query.phpstub
@@ -14,7 +14,7 @@ namespace yii\db;
  * @template TModel
  * @template TMultiple
  * @method iterable<mixed, TModel> each(int $batchSize = 100, ?Connection $db = null)
- * @method TModel|null one()
- * @method array<mixed, TModel> all()
+ * @method TModel|null one($db = null)
+ * @method array<mixed, TModel> all($db = null)
  */
 class Query {}