X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/8af012bc2aaa0dacc64c0f69e78b34a2e0472119..refs/pull/806/head:/app/Activity.php

diff --git a/app/Activity.php b/app/Activity.php
index bbbe5e893..c01da1f6c 100644
--- a/app/Activity.php
+++ b/app/Activity.php
@@ -1,8 +1,6 @@
 <?php
 
-namespace Oxbow;
-
-use Illuminate\Database\Eloquent\Model;
+namespace BookStack;
 
 /**
  * @property string  key
@@ -15,15 +13,13 @@ class Activity extends Model
 
     /**
      * Get the entity for this activity.
-     * @return bool
      */
     public function entity()
     {
-        if ($this->entity_id) {
-            return $this->morphTo('entity')->first();
-        } else {
-            return false;
+        if ($this->entity_type === '') {
+            $this->entity_type = null;
         }
+        return $this->morphTo('entity');
     }
 
     /**
@@ -32,7 +28,7 @@ class Activity extends Model
      */
     public function user()
     {
-        return $this->belongsTo('Oxbow\User');
+        return $this->belongsTo(User::class);
     }
 
     /**
@@ -49,8 +45,8 @@ class Activity extends Model
      * @param $activityB
      * @return bool
      */
-    public function isSimilarTo($activityB) {
-        return [$this->key, $this->entitiy_type, $this->entitiy_id] === [$activityB->key, $activityB->entitiy_type, $activityB->entitiy_id];
+    public function isSimilarTo($activityB)
+    {
+        return [$this->key, $this->entity_type, $this->entity_id] === [$activityB->key, $activityB->entity_type, $activityB->entity_id];
     }
-
 }