fix(tests): add waits in direct webdriver calls