diff --git a/code/TestRegistryPage.php b/code/TestRegistryPage.php
index 76bd6db..020a9b4 100644
--- a/code/TestRegistryPage.php
+++ b/code/TestRegistryPage.php
@@ -6,7 +6,7 @@ use SilverStripe\Registry\RegistryPage;
 use SilverStripe\Security\DefaultAdminService;
 use SilverStripe\Security\Member;
 
-if (!class_exists(RegistryPage::class)) {
+if (class_exists(RegistryPage::class)) {
     class TestRegistryPage extends RegistryPage
     {
         public function requireDefaultRecords()
diff --git a/code/elemental/ElementContentExtension.php b/code/elemental/ElementContentExtension.php
index d3e7273..06145f7 100644
--- a/code/elemental/ElementContentExtension.php
+++ b/code/elemental/ElementContentExtension.php
@@ -2,6 +2,8 @@
 
 namespace SilverStripe\FrameworkTest\Elemental\Extension;
 
+use SilverStripe\Assets\File;
+use SilverStripe\CMS\Model\SiteTree;
 use SilverStripe\Core\Extension;
 use SilverStripe\Forms\FieldList;
 use SilverStripe\ORM\ValidationResult;
@@ -20,6 +22,11 @@ class ElementContentExtension extends Extension
         'MyInt' => 'Int',
     ];
 
+    private static $has_one = [
+        'MyPage' => SiteTree::class,
+        'MyFile' => File::class,
+    ];
+
     public function validate(ValidationResult $result)
     {
         if ($this->owner->Title == 'x') {
@@ -35,7 +42,7 @@ class ElementContentExtension extends Extension
 
     public function updateCMSCompositeValidator(CompositeValidator $compositeValidator)
     {
-        $compositeValidator->addValidator(new RequiredFields(['Title']));
+        $compositeValidator->addValidator(new RequiredFields(['Title', 'MyPageID', 'MyFile']));
     }
 
     public function updateCMSFields(FieldList $fields)