diff --git a/BasicStateCodelab/app/src/main/java/jik/compose/basicstatecodelab/WaterCounter.kt b/BasicStateCodelab/app/src/main/java/jik/compose/basicstatecodelab/WaterCounter.kt
index 0103bf4..9a15bc2 100644
--- a/BasicStateCodelab/app/src/main/java/jik/compose/basicstatecodelab/WaterCounter.kt
+++ b/BasicStateCodelab/app/src/main/java/jik/compose/basicstatecodelab/WaterCounter.kt
@@ -4,14 +4,14 @@ import androidx.compose.foundation.layout.Column
 import androidx.compose.foundation.layout.padding
 import androidx.compose.material.Button
 import androidx.compose.material.Text
-import androidx.compose.runtime.Composable
+import androidx.compose.runtime.*
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.unit.dp
 
 @Composable
 fun WaterCounter(modifier: Modifier = Modifier) {
     Column(modifier = modifier.padding(16.dp)) {
-        var count = 0
+        var count by remember { mutableStateOf(0) }
         Text(text = "You've had $count glasses")
         Button(onClick = { count++ }, modifier = Modifier.padding(top = 8.dp)) {
             Text(text = "Add one")