I think that you must wipe cache and dalvik but wipe data is strongly recomended travelling from Android 8 to 9.
I passed from global android 8 to EU android 9 with all wipes and all right.
A mí me pasa exactamente lo mismo.
Si lo bloqueo y acto seguido lo desbloqueo con el botón sí que me reconoce pero con las notificaciones en la pantalla de bloqueo no funciona.
Enviado desde mi MI MAX 3 mediante Tapatalk