Xiaomi Pad 6 aftermath of Hyper OS 2 update

After updating my Pad 6, the HYPEROS2. My Xiaomi keyboard keeps on disconnecting. I need to restart/reboot my tablet in order to connect again on my keyboard. P.S Official xiaomi pad 6 keyboard