From 1fe4827be08a31f06cb8314837b80db1fc696a9b Mon Sep 17 00:00:00 2001
From: Alban Jubert <alban@lahautesociete.com>
Date: Sat, 3 Nov 2018 22:50:26 +0100
Subject: [PATCH] Fix for issue #23

---
 src/SaveRelationsBehavior.php | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/SaveRelationsBehavior.php b/src/SaveRelationsBehavior.php
index ce261fc..5af0809 100644
--- a/src/SaveRelationsBehavior.php
+++ b/src/SaveRelationsBehavior.php
@@ -288,6 +288,10 @@ class SaveRelationsBehavior extends Behavior
         }
         if (($relationModel instanceof BaseActiveRecord) && is_array($data)) {
             $relationModel->setAttributes($data);
+            if ($relationModel->hasMethod('loadRelations')) {
+                $relationModel->loadRelations($data);
+            }
+
         }
         return $relationModel;
     }
@@ -738,7 +742,7 @@ class SaveRelationsBehavior extends Behavior
     }
 
     /**
-     * @param $relationName
+     * @param $relationName string
      * @return mixed
      * @throws InvalidConfigException
      */
@@ -779,7 +783,7 @@ class SaveRelationsBehavior extends Behavior
 
     /**
      * Returns the old value of the named relation.
-     * @param $relationName The relations name as defined in the behavior `relations` parameter
+     * @param $relationName string The relations name as defined in the behavior `relations` parameter
      * @return mixed
      */
     public function getOldRelation($relationName)
@@ -804,7 +808,7 @@ class SaveRelationsBehavior extends Behavior
 
     /**
      * Mark a relation as dirty
-     * @param $relationName
+     * @param $relationName string
      * @return bool Whether the operation succeeded.
      */
     public function markRelationDirty($relationName)