From 17733aa84ad4852bd5a5e8b157790d0993485d9c Mon Sep 17 00:00:00 2001
From: Rastislav Brandobur <brandobur@tjbw.sk>
Date: Mon, 11 Sep 2023 10:58:11 +0200
Subject: [PATCH] added missing returns (fluent setter)

---
 src/ORM/FieldType/DBClassName.php  | 2 ++
 src/ORM/FieldType/DBPrimaryKey.php | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/ORM/FieldType/DBClassName.php b/src/ORM/FieldType/DBClassName.php
index 0b5943593..48229afda 100644
--- a/src/ORM/FieldType/DBClassName.php
+++ b/src/ORM/FieldType/DBClassName.php
@@ -151,6 +151,8 @@ class DBClassName extends DBEnum
         if ($record instanceof DataObject) {
             $this->record = $record;
         }
+
+        return $this;
     }
 
     public function getDefault()
diff --git a/src/ORM/FieldType/DBPrimaryKey.php b/src/ORM/FieldType/DBPrimaryKey.php
index 13c047a5e..8132bfa3f 100644
--- a/src/ORM/FieldType/DBPrimaryKey.php
+++ b/src/ORM/FieldType/DBPrimaryKey.php
@@ -68,5 +68,7 @@ class DBPrimaryKey extends DBInt
         if ($record instanceof DataObject) {
             $this->object = $record;
         }
+
+        return $this;
     }
 }