added awaitility to fix flickering test