diff --git a/app/src/test/java/com/x8bit/bitwarden/ui/vault/feature/itemlisting/VaultItemListingViewModelTest.kt b/app/src/test/java/com/x8bit/bitwarden/ui/vault/feature/itemlisting/VaultItemListingViewModelTest.kt
index 38251688eb..84713ba6a0 100644
--- a/app/src/test/java/com/x8bit/bitwarden/ui/vault/feature/itemlisting/VaultItemListingViewModelTest.kt
+++ b/app/src/test/java/com/x8bit/bitwarden/ui/vault/feature/itemlisting/VaultItemListingViewModelTest.kt
@@ -5,6 +5,7 @@ import app.cash.turbine.test
 import com.x8bit.bitwarden.R
 import com.x8bit.bitwarden.data.platform.repository.model.DataState
 import com.x8bit.bitwarden.data.vault.datasource.sdk.model.createMockCipherView
+import com.x8bit.bitwarden.data.vault.datasource.sdk.model.createMockCollectionView
 import com.x8bit.bitwarden.data.vault.datasource.sdk.model.createMockFolderView
 import com.x8bit.bitwarden.data.vault.repository.VaultRepository
 import com.x8bit.bitwarden.data.vault.repository.model.VaultData
@@ -100,6 +101,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                             ),
                         ),
                         folderViewList = listOf(createMockFolderView(number = 1)),
+                        collectionViewList = listOf(createMockCollectionView(number = 1)),
                     ),
                 ),
             )
@@ -126,6 +128,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                     data = VaultData(
                         cipherViewList = emptyList(),
                         folderViewList = emptyList(),
+                        collectionViewList = emptyList(),
                     ),
                 ),
             )
@@ -144,6 +147,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                     data = VaultData(
                         cipherViewList = listOf(createMockCipherView(number = 1)),
                         folderViewList = listOf(createMockFolderView(number = 1)),
+                        collectionViewList = listOf(createMockCollectionView(number = 1)),
                     ),
                 ),
             )
@@ -175,6 +179,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = listOf(createMockCipherView(number = 1, isDeleted = false)),
                     folderViewList = listOf(createMockFolderView(number = 1)),
+                    collectionViewList = listOf(createMockCollectionView(number = 1)),
                 ),
             ),
         )
@@ -200,6 +205,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = listOf(createMockCipherView(number = 1)),
                     folderViewList = listOf(createMockFolderView(number = 1)),
+                    collectionViewList = listOf(createMockCollectionView(number = 1)),
                 ),
             ),
         )
@@ -219,6 +225,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = listOf(createMockCipherView(number = 1)),
                     folderViewList = listOf(createMockFolderView(number = 1)),
+                    collectionViewList = listOf(createMockCollectionView(number = 1)),
                 ),
             ),
         )
@@ -258,6 +265,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = listOf(createMockCipherView(number = 1, isDeleted = false)),
                     folderViewList = listOf(createMockFolderView(number = 1)),
+                    collectionViewList = listOf(createMockCollectionView(number = 1)),
                 ),
                 error = IllegalStateException(),
             ),
@@ -284,6 +292,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = emptyList(),
                     folderViewList = emptyList(),
+                    collectionViewList = emptyList(),
                 ),
                 error = IllegalStateException(),
             ),
@@ -306,6 +315,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = listOf(createMockCipherView(number = 1, isDeleted = true)),
                     folderViewList = listOf(createMockFolderView(number = 1)),
+                    collectionViewList = listOf(createMockCollectionView(number = 1)),
                 ),
                 error = IllegalStateException(),
             ),
@@ -346,9 +356,11 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
         mutableVaultDataStateFlow.tryEmit(
             value = DataState.NoNetwork(
                 data = VaultData(
-                cipherViewList = listOf(createMockCipherView(number = 1, isDeleted = false)),
-                folderViewList = listOf(createMockFolderView(number = 1)),
-            )),
+                    cipherViewList = listOf(createMockCipherView(number = 1, isDeleted = false)),
+                    folderViewList = listOf(createMockFolderView(number = 1)),
+                    collectionViewList = listOf(createMockCollectionView(number = 1)),
+                ),
+            ),
         )
 
         val viewModel = createVaultItemListingViewModel()
@@ -372,6 +384,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = emptyList(),
                     folderViewList = emptyList(),
+                    collectionViewList = emptyList(),
                 ),
             ),
         )
@@ -393,6 +406,7 @@ class VaultItemListingViewModelTest : BaseViewModelTest() {
                 data = VaultData(
                     cipherViewList = listOf(createMockCipherView(number = 1, isDeleted = true)),
                     folderViewList = listOf(createMockFolderView(number = 1)),
+                    collectionViewList = listOf(createMockCollectionView(number = 1)),
                 ),
             ),
         )