[Android]Jetpack ComposeのComposableのModifierを条件によって切り替えたい

例えば

val modifier = 
if (isMax) {
         Modifier.fillMaxWidth()
                 .fillMaxHeight()
} else {
         Modifier.fillMaxWidth()
                 .wrapContentHeight()
}

AnyComposable(modifier = modifier)

というようにheightの部分だけ条件によって変更したい時
以下のコードだと.applyの中身はmodifierとして適用されない

AnyComposable(modifier = 
         Modifier.fillMaxWidth()
                 .apply {
                     if (isMax) {
                          fillMaxHeight()
                     } else {
                          wrapContentHeight()
                     }
                 }
)

thenを使う
以下が正解

AnyComposable(modifier = 
         Modifier.fillMaxWidth()
                 .then (
                     if (isMax) {
                          Modifier.fillMaxHeight()
                     } else {
                          Modofier.wrapContentHeight()
                     }
                 )
)