replace self.driver.connection with self