diff --git a/src/Traits/FindMethodTrait.php b/src/Traits/FindMethodTrait.php index 0d7a7dcdb..4db49fdd7 100644 --- a/src/Traits/FindMethodTrait.php +++ b/src/Traits/FindMethodTrait.php @@ -61,11 +61,12 @@ protected function _findRecord(?string $id, Subject $subject): EntityInterface [$finder, $options] = $this->_extractFinder(); $query = $repository->find($finder, $options); + $primaryKey = $this->getConfig('primaryKey', $repository->getPrimaryKey()); /** * @psalm-suppress PossiblyInvalidArgument * @psalm-suppress InvalidArrayOffset */ - $query->where([current($query->aliasField($repository->getPrimaryKey())) => $id]); + $query->where([current($query->aliasField($primaryKey)) => $id]); $subject->set([ 'repository' => $repository,