diff --git a/tests/test_clamav.py b/tests/test_clamav.py
index 8d3c030..3911b4d 100644
--- a/tests/test_clamav.py
+++ b/tests/test_clamav.py
@@ -37,7 +37,7 @@ def _upload_file(self, url, file_path):
         self.driver.get("%s%s" % (self.live_server_url, url))
         self.driver.find_element(By.ID, "id_abstracts_upload").send_keys(file_path)
         self.driver.find_element(By.ID, "upload_button").click()
-        WebDriverWait(self.driver, timeout=30, poll_frequency=0.5).until(lambda x: x.find_element_by_id('id_include_child_nodes_1'))
+        WebDriverWait(self.driver, timeout=30, poll_frequency=0.5).until(lambda x: x.find_element((By.ID, "id_include_child_nodes_1")))
 
     def _assert_file_upload(self, url, file_path):
         logger.debug('_assert_file_upload %s %s ' % (url, file_path))