diff --git a/src/Traits/HasEmbeddings.php b/src/Traits/HasEmbeddings.php
index 3655fce..b4c3faa 100644
--- a/src/Traits/HasEmbeddings.php
+++ b/src/Traits/HasEmbeddings.php
@@ -6,6 +6,9 @@
 use Ada\Jobs\EmbedJob;
 use Ada\Models\Embedding;
 
+/**
+ * @mixin \Illuminate\Database\Eloquent\Model
+ */
 trait HasEmbeddings
 {
     public function embed(string $key, string $content): bool
@@ -24,7 +27,7 @@ public function embed(string $key, string $content): bool
         foreach ($embeddings as $embedding) {
             $embedding->key = $key;
             $embedding->embeddable_type = static::class;
-            $embedding->embeddable_id = $this->id;
+            $embedding->embeddable_id = $this->getKey();
 
             $embedding->save();