Melanie/Ajay | #0000 | Fixing flaky ft test - wait for dialog to close