X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/bc1302a8d8c9c7facb16b28e1cf1da4409e7ede0..refs/pull/806/head:/tests/UserProfileTest.php

diff --git a/tests/UserProfileTest.php b/tests/UserProfileTest.php
index 4636fc76c..c1e254090 100644
--- a/tests/UserProfileTest.php
+++ b/tests/UserProfileTest.php
@@ -97,24 +97,23 @@ class UserProfileTest extends BrowserKitTest
 
     public function test_books_view_is_list()
     {
-        $editor = $this->getEditor([
-          'books_view_type' => 'list'
-        ]);
+        $editor = $this->getEditor();
+        setting()->putUser($editor, 'books_view_type', 'list');
 
         $this->actingAs($editor)
             ->visit('/books')
             ->pageNotHasElement('.featured-image-container')
-            ->pageHasElement('.entity-list-item');
+            ->pageHasElement('.content .entity-list-item');
     }
 
     public function test_books_view_is_grid()
     {
-        $editor = $this->getEditor([
-          'books_view_type' => 'grid'
-        ]);
+        $editor = $this->getEditor();
+        setting()->putUser($editor, 'books_view_type', 'grid');
 
         $this->actingAs($editor)
             ->visit('/books')
             ->pageHasElement('.featured-image-container');
     }
+
 }