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;
     }
 }