X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/a3ead5062acc169ae3486d90ac2befe3db86bfe6..refs/pull/5280/head:/tests/Commands/ClearActivityCommandTest.php

diff --git a/tests/Commands/ClearActivityCommandTest.php b/tests/Commands/ClearActivityCommandTest.php
index 71baa0ca6..a3d7456b6 100644
--- a/tests/Commands/ClearActivityCommandTest.php
+++ b/tests/Commands/ClearActivityCommandTest.php
@@ -2,8 +2,7 @@
 
 namespace Tests\Commands;
 
-use BookStack\Actions\ActivityType;
-use BookStack\Entities\Models\Page;
+use BookStack\Activity\ActivityType;
 use BookStack\Facades\Activity;
 use Illuminate\Support\Facades\Artisan;
 use Illuminate\Support\Facades\DB;
@@ -14,14 +13,13 @@ class ClearActivityCommandTest extends TestCase
     public function test_clear_activity_command()
     {
         $this->asEditor();
-        /** @var Page $page */
-        $page = Page::query()->first();
+        $page = $this->entities->page();
         Activity::add(ActivityType::PAGE_UPDATE, $page);
 
         $this->assertDatabaseHas('activities', [
             'type'      => 'page_update',
-            'entity_id' => $page->id,
-            'user_id'   => $this->getEditor()->id,
+            'loggable_id' => $page->id,
+            'user_id'   => $this->users->editor()->id,
         ]);
 
         DB::rollBack();