diff --git a/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt b/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt
index 2d87c3be6e..4d271eabde 100644
--- a/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt
+++ b/examples/llama.android/app/src/main/java/com/example/llama/revamp/ui/screens/ModelSelectionScreen.kt
@@ -68,6 +68,9 @@ fun ModelSelectionScreen(
                         ModelCardExpandable(
                             model = model,
                             isSelected = if (model == preselectedModel) true else null,
+                            onSelected = { selected ->
+                                if (!selected) viewModel.resetSelection()
+                            },
                             isExpanded = model == preselectedModel,
                             onExpanded = { expanded ->
                                 viewModel.preselectModel(model, expanded)