Update Entity.php

PHPixie · May 15, 2017 · 5f5d057 · 5f5d057
1 parent 2554d00
commit 5f5d057
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/PHPixie/ORM/Wrappers/Model/Entity.php b/src/PHPixie/ORM/Wrappers/Model/Entity.php
@@ -67,6 +67,11 @@ public function __set($name, $value)
 
     public function __call($method, $params)
     {
-        return $this->entity->__call($method, $params);
+        $result = $this->entity->__call($method, $params);
+        if($result === $this->entity) {
+            return $this;
+        }
+
+        return $result;
     }
 }