Ревью: ожидания в Selenium-тесте

from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time

def test_payment(driver):
    driver.implicitly_wait(10)
    driver.get("https://stage.shop.local/checkout")
    time.sleep(5)
    driver.find_element(By.ID, "card").send_keys("4111111111111111")
    WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "pay")))
    driver.find_element(By.ID, "pay").click()
    time.sleep(3)
    assert "Оплачено" in driver.page_source