diff --git a/src/Plugin/RulesAction/EntityCreateDeriver.php b/src/Plugin/RulesAction/EntityCreateDeriver.php
index aa096ae..45c0674 100644
--- a/src/Plugin/RulesAction/EntityCreateDeriver.php
+++ b/src/Plugin/RulesAction/EntityCreateDeriver.php
@@ -96,6 +96,11 @@ class EntityCreateDeriver extends DeriverBase implements ContainerDeriverInterfa
         if ($type_definition instanceof DataReferenceDefinitionInterface) {
           $type_definition->getTargetDefinition();
         }
+
+        if (!is_object($type_definition) || !method_exists($type_definition, 'getDataType')) {
+          continue;
+        }
+
         $type = $type_definition->getDataType();
 
         $is_bundle = ($field_name == $bundle_key);