Fixed what needed fixing and squished some bugs. 🐛

Author: abr4xas

src/Models/Lead.php

@@ -2,10 +2,11 @@
 
 namespace Tepuilabs\SimpleCrm\Models;
 
+use Illuminate\Database\Eloquent\Factories\HasFactory;
 use Illuminate\Database\Eloquent\Model;
-use Tepuilabs\SimpleCrm\Models\Enums\LeadStatus;
 use Illuminate\Database\Eloquent\Relations\HasMany;
-use Illuminate\Database\Eloquent\Factories\HasFactory;
+use Illuminate\Database\Eloquent\Relations\MorphMany;
+use Tepuilabs\SimpleCrm\Models\Enums\LeadStatus;
 
 /**
  * @url https://www.hipb2b.com/blog/lead-prospect-whats-difference
@@ -29,16 +30,16 @@ class Lead extends Model
     protected $casts = [
         'status' => LeadStatus::class,
         'type' => LeadStatus::class,
-	];
+    ];
 
     /**
      * Notes relationship
      *
-     * @return HasMany
+     * @return MorphMany
      */
-    public function notes(): HasMany
+    public function notes(): MorphMany
     {
-        return $this->hasMany(\Tepuilabs\SimpleCrm\Models\Note::class);
+        return $this->morphMany(\Tepuilabs\SimpleCrm\Models\Note::class, 'commentable');
     }
 
     /**

src/Models/Note.php

@@ -32,6 +32,11 @@ public function author(): MorphTo
         return $this->morphTo();
     }
 
+    public function commentable(): MorphTo
+    {
+        return $this->morphTo();
+    }
+
     public function lead(): BelongsTo
     {
         return $this->belongsTo(\Tepuilabs\SimpleCrm\Models\Lead::class);