AndroidView focus in Jetpack Compose

I want to keep track of focus in EditText in an AndroidView composable like in legacy UI. I've tried onFocusChange() in EditText but that doesn't get triggered. Also Modifier.OnFocusChange() doesn't get triggered either. I don't want to use TextField cause I have some operations that have to be done in native EditText.