Try to fix previous commit

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -122,14 +122,14 @@ def run_notebook(driver, notebook_area, args):
         notebook_area.send_keys(Keys.SHIFT, Keys.ENTER)
         # This sleep is there is allow time for the box for standard input
         # to appear, if it needs to after executing the cell
-        time.sleep(0.2)
+        time.sleep(0.1)
         if not cell_is_waiting_for_input(driver):
             wait_for_idle_status(driver, current_cell, start_time, args.timeout)
-            next_cell = current_cell.find_elements(
+            next_cell = current_cell.find_element(
                 By.XPATH,
-                "following-sibling::div[contains(@class,'jp-Notebook-cell')][1]"
+                "following-sibling::div[contains(@class,'jp-Notebook-cell')][1]",
             )
-            current_cell = next_cell[0]
+            current_cell = next_cell
 
 def download_notebook(driver):
     """This function is used to download the notebook currently open."""