diff --git a/app/src/test/java/android/template/ui/mymodel/MyModelViewModelTest.kt b/app/src/test/java/android/template/ui/mymodel/MyModelViewModelTest.kt
index 781e6c6..2b10768 100644
--- a/app/src/test/java/android/template/ui/mymodel/MyModelViewModelTest.kt
+++ b/app/src/test/java/android/template/ui/mymodel/MyModelViewModelTest.kt
@@ -62,8 +62,9 @@ class MyModelViewModelTest {
     @Test
     fun uiState_onItemSaved_isDisplayed() = runTest {
         val viewModel = MyModelViewModel(FakeMyModelRepository())
-        assertEquals(MyModelUiState.Loading, viewModel.uiState.value)
         assertEquals(MyModelUiState.Success(emptyList()), viewModel.uiState.first())
+        viewModel.addMyModel("Compose")
+        assertEquals(MyModelUiState.Success(listOf("Compose")), viewModel.uiState.first())
     }
 }