- Some overdue modification made to allow more flexibility

src/Repository.php

@@ -65,7 +65,7 @@ class Repository
         return $primaryKey ? $this->loadOneFromField($primaryKey, $value) : $this->wherePrimaryKey($value)->loadOne();
     }
 
-    public function loadAll() : EntityCollection
+    public function loadAll () : EntityCollection
     {
         return $this->selectSqlQuery()->collectionFromQuery();
     }
@@ -590,7 +590,7 @@ class Repository
 
         $pkField = key($primaryKeyField);
 
-        return $this->where($primaryKeyField[$pkField]->name ?? $pkField, $value);
+        return $this->where($this->entityClass::field($primaryKeyField[$pkField]->name ?? $pkField), $value);
     }
 
     public function withJoin(string|array $fields, array $options = []) : self