diff --git a/src/Services/DataTable.php b/src/Services/DataTable.php
index 10f94d5..0b791d5 100644
--- a/src/Services/DataTable.php
+++ b/src/Services/DataTable.php
@@ -681,11 +681,11 @@ public function addScopes(array $scopes): static
     /**
      * Set a custom class attribute.
      *
-     * @param  array|string  $key
-     * @param  array|string|null  $value
+     * @param  array|string $key
+     * @param  mixed|null $value
      * @return $this
      */
-    public function with(array|string $key, array|string $value = null): static
+    public function with(array|string $key, mixed $value = null): static
     {
         if (is_array($key)) {
             $this->attributes = array_merge($this->attributes, $key);