diff --git a/src/Facades/Ui.php b/src/Facades/Ui.php
index 486e5da..ec9f7da 100644
--- a/src/Facades/Ui.php
+++ b/src/Facades/Ui.php
@@ -18,7 +18,7 @@
  * @method static ?Theme getTheme(string $themeKey)
  * @method static bool hasTheme(string $themeKey)
  * @method static \Playground\Blade\Ui saveTheme(Theme $theme)
- * @method static \Playground\Blade\Ui setTheme(string $themeKey = '')
+ * @method static \Playground\Blade\Ui setTheme(string $themeKey = '', bool $save = true)
  */
 class Ui extends Facade
 {