Tweak out-of-focus wait code in demo