Fix unit test

1 files changed, 2 insertions, 2 deletions

diff --git a/api/preference_test.go b/api/preference_test.go
index eaa92fe47..4498ad7c3 100644
--- a/api/preference_test.go
+++ b/api/preference_test.go
@@ -48,7 +48,7 @@ func TestGetAllPreferences(t *testing.T) {
 
 	if result, err := Client.GetAllPreferences(); err != nil {
 		t.Fatal(err)
-	} else if data := result.Data.(model.Preferences); len(data) != 3 {
+	} else if data := result.Data.(model.Preferences); len(data) != 4 {
 		t.Fatal("received the wrong number of preferences")
 	} else if !((data[0] == preferences1[0] && data[1] == preferences1[1]) || (data[0] == preferences1[1] && data[1] == preferences1[0])) {
 		for i := 0; i < 3; i++ {
@@ -63,7 +63,7 @@ func TestGetAllPreferences(t *testing.T) {
 	// note that user2 will automatically have a preference set for them to show user1 for direct messages
 	if result, err := Client.GetAllPreferences(); err != nil {
 		t.Fatal(err)
-	} else if data := result.Data.(model.Preferences); len(data) != 1 {
+	} else if data := result.Data.(model.Preferences); len(data) != 2 {
 		t.Fatal("received the wrong number of preferences")
 	}
 }