Picture-in-Picture with custom ratio doesn't work correctly on Android 11

[paulchernenko]

I'm trying to run the demo app with a custom ratio on Pixel devices with Android 11 or on the emulator with Android 11 modifying only this code

enterPictureInPictureMode(
    PictureInPictureParams.Builder().setAspectRatio(
        Rational(2, 3)
    ).build()
)

As a result the app enters the picture-in-picture mode correctly, but when I try to exit it by clicking on the window it doesn't expand to the full screen, but the UI changes like the picture-in-picture mode is disabled. What helps to open the app is opening the application drawer or notification center. This issue doesn't happen every time, but most of the times.
Also it doesn't happen on emulators with Android 10 or Android S, and it doesn't happen on devices other than Pixel. For example, on Samsung with Android 11 it is working fine.