move assertion inside driver.quit()