Samsung is rolling out a new update for Always On Display, pushing the app to version 5.2.05.8. It fixes a couple of bugs related to Dark Mode and Bluetooth-connected keyboards, and also claims to improve the layouts for some clock styles.
Before this update, Always On Display had an issue whereby the background would turn gray when using a Bluetooth keyboard. AOD settings were also difficult to see while Dark Mode was enabled on certain themes. These bugs have now been addressed with version 5.2.05.8.
You don't need One UI 2.1 for this update
The update is rolling out now, and we can confirm that it is available at least on the Galaxy Note 10 (running One UI 2.0 and not the latest One UI 2.1 update). But the new Always On Display version should be pushed to more Samsung devices as we speak.
You can check which Always On Display version you're running by opening Settings, navigating to Lock screen » Always On Display (don't tap the toggle, but the text itself), and tapping About Always On Display. If the update hasn't reached your phone yet, you may have to wait for a few more days. You can also launch the Galaxy Store on your phone, tap the ‘hamburger' menu in the upper-left corner, and check if the new version is waiting for you in the Updates category.