diff --git a/app/src/test/java/awais/instagrabber/viewmodels/ProfileFragmentViewModelTest.kt b/app/src/test/java/awais/instagrabber/viewmodels/ProfileFragmentViewModelTest.kt
index fb851c7e..2c425892 100644
--- a/app/src/test/java/awais/instagrabber/viewmodels/ProfileFragmentViewModelTest.kt
+++ b/app/src/test/java/awais/instagrabber/viewmodels/ProfileFragmentViewModelTest.kt
@@ -279,6 +279,7 @@ internal class ProfileFragmentViewModelTest {
         while (profile.status == Resource.Status.LOADING) {
             profile = viewModel.profile.getOrAwaitValue()
         }
+        assertEquals(true, viewModel.isFavorite.getOrAwaitValue())
         assertTrue(updateFavoriteCalled)
     }
 }
\ No newline at end of file