diff --git a/EActiveRecordRelationBehavior.php b/EActiveRecordRelationBehavior.php
index 15931ea..484660e 100644
--- a/EActiveRecordRelationBehavior.php
+++ b/EActiveRecordRelationBehavior.php
@@ -297,7 +297,8 @@ protected function populateBelongsToAttribute($name, $relation, $beforeSave = tr
 		}
 
 		// @todo add support for composite primary keys
-		if (!is_array($pk)) {
+		// @todo add support for relation keys mapping
+		if (!is_array($pk) && !is_array($relation[2])) {
 			$this->owner->setAttribute($relation[2], $pk);
 		}
 	}