Fix driver.run_once