diff --git a/src/UserInterface/Eloquent/Concerns/HasSchemalessAttributes.php b/src/UserInterface/Eloquent/Concerns/HasSchemalessAttributes.php
index e699cdadf..72315624c 100644
--- a/src/UserInterface/Eloquent/Concerns/HasSchemalessAttributes.php
+++ b/src/UserInterface/Eloquent/Concerns/HasSchemalessAttributes.php
@@ -5,8 +5,11 @@
 namespace ARKEcosystem\Foundation\UserInterface\Eloquent\Concerns;
 
 use Illuminate\Database\Eloquent\Builder;
-use Illuminate\Database\Eloquent\Model;
+use Spatie\SchemalessAttributes\SchemalessAttributes;
 
+/**
+ * @property SchemalessAttributes $extra_attributes
+ */
 trait HasSchemalessAttributes
 {
     /**
@@ -97,7 +100,7 @@ public function fillMetaAttributes(array $attributes): self
      *
      * @return self
      */
-    public static function updateOrCreateWithMeta(array $attributes, array $values): Model
+    public static function updateOrCreateWithMeta(array $attributes, array $values): self
     {
         $model = static::withExtraAttributes($attributes)->first();