I'm aware that Power Saving disables AOD while in use but AOD continues to be disabled even after turning off Power Saving. I have to Manually go back into AOD settings and turn it back on. Not sure if this is an S26U issue or OneUI issue (was an issue with OneUI 8.5 as well).
