diff --git a/code/ElementsExtension.php b/code/ElementsExtension.php
index <HASH>..<HASH> 100644
--- a/code/ElementsExtension.php
+++ b/code/ElementsExtension.php
@@ -246,7 +246,7 @@ class ElementsExtension extends DataExtension
             $tabName = "Root.{$relationName}";
             $label = _t("Element_Relations.{$relationName}", $relationName);
             $fields->addFieldToTab($tabName,
-                $gridfieldItems = GridField::create(
+                $gridField = GridField::create(
                     $relationName,
                     $label,
                     $this->owner->getItemsByRelation($relationName),
@@ -254,6 +254,11 @@ class ElementsExtension extends DataExtension
                 )
             );
 
+            if (count($elementClasses) == 1)
+            {
+                $gridField->setModelClass($elementClasses[0]);
+            }
+
             $fields
                 ->findOrMakeTab($tabName)
                 ->setTitle($label)
set proper model class for one element only