From 0ffec5e15aa4da62c44a165347e715041d8c7ec1 Mon Sep 17 00:00:00 2001
From: Sad Ellie <openthiswhitedoor@gmail.com>
Date: Tue, 23 Aug 2022 22:24:59 +0300
Subject: [PATCH] No selected unit group bugfix

---
 .../com/sadellie/unitto/screens/second/SecondViewModel.kt     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/com/sadellie/unitto/screens/second/SecondViewModel.kt b/app/src/main/java/com/sadellie/unitto/screens/second/SecondViewModel.kt
index 772427c0..9ed808a6 100644
--- a/app/src/main/java/com/sadellie/unitto/screens/second/SecondViewModel.kt
+++ b/app/src/main/java/com/sadellie/unitto/screens/second/SecondViewModel.kt
@@ -46,7 +46,9 @@ class SecondViewModel @Inject constructor(
     private val _uiStateFlow = MutableStateFlow(SecondScreenUIState())
 
     val mainFlow = combine(_uiStateFlow, unitGroupsRepository.shownUnitGroups) { uiState, shown ->
-        return@combine uiState.copy(shownUnitGroups = shown)
+        val newState = uiState.copy(shownUnitGroups = shown)
+        _uiStateFlow.value = newState
+        return@combine newState
     }
         .stateIn(
             scope = viewModelScope,