MutableStateOf não está sendo reinicializado no Jetpack Compose

Abaixo está uma função simples que pode ser composta, quando eu logo o objeto pd ele imprime o valor atual do campo retido (booleano). Mas depois de inicializar a variável isChecked o valor é sempre! Portanto, o estado do meu switch está sempre incorreto.

@Composable
fun DocumentModalSheet( pd: PrintDocument,
onSwitchSelected: (id: Long, status: Boolean) -> Unit,
) {
Log.i("DocumentModalSheet", "DocumentModalSheet: ${pd.retained}")
// This prints the correct value of pd.retained
var isChecked by remember {
    mutableStateOf(pd.retained)
}
Log.i("DocumentModalSheet", "DocumentModalSheet: $isChecked") 
// This prints the incorrect value of pd.retained       
Column(
    modifier = Modifier.fillMaxWidth()
) {
    Switch(checked = isChecked, onCheckedChange = {
        isChecked = it,
        onSwitchSelected(pd.id, it)
    } )
}}

É assim que estou chamando a função acima do elemento que pode ser composto PrintDocumentScreen. Quando o lambda do DocumentCard é chamado, estou atualizando a variável clickedPD e mostrando a planilha modal.

@Composable
fun PrintDocumentScreen(viewModel: PrintDocumentViewModel) {
val pdListFlow = viewModel.pdListFlow.collectAsState()
var clickedPD by remember { mutableStateOf(PrintDocument()) }
val modalSheetState = rememberModalBottomSheetState(..)

ModalBottomSheetLayout(
    sheetContent = {
        DocumentModalSheet(clickedPD){
         viewModel.doSomething(id, status)
        }
    },
    content = {
        Scaffold(modifier = Modifier.fillMaxSize()) { paddingValues ->
            Column(modifier = Modifier.fillMaxSize().padding(paddingValues)
            ) {
                pdListFlow.value?.let { it ->
                    when (it) {
                        is Resource.Success -> {
                            if (it.data.printDocuments.isNotEmpty()) {
                                LazyColumn(modifier = Modifier.fillMaxSize()) {
                                    items(it.data.printDocuments.size) { index ->
                                        DocumentCard(it.data.printDocuments[index]) {
                                            clickedPD = it
                                            coroutineScope.launch {
                                                modalSheetState.show()
                                            }
                                        }
                                    }
                                }

                            } else {
                                NoDocumentPresent()
                            }
                        }
                    }
                }
            }}})}

O que eu tentei até agora...

Tentei içar a parte de inicialização isChecked para PrintDocumentScreen, mas não ajudou.

Também usei derivadoStateOf() para inicializar a variável isChecked no PrintDocumentScreen

var clickedPD by remember { mutableStateOf(PrintDocument()) }
val isChecked by remember {
    derivedStateOf { clickedPD.retained }
}

Percebi que o estado do switch é compartilhado por todos os DocumentCard(s), pois quando eu verifico um switch, escondo a folha modal, toco em outro cartão, o switch já está selecionado.