diff --git a/test/test.py b/test/test.py
index 6e45b34..52b62f1 100644
--- a/test/test.py
+++ b/test/test.py
@@ -29,7 +29,7 @@ def download_widget_image(self):
     time.sleep(3)
     self.driver.save_screenshot("widget-01.png")
 
-    body = self.driver.find_element(By.TAG_NAME, 'body')
+    body = self.driver.find_element(By.CLASS_NAME, 'bands-widget-all')
     body.send_keys(Keys.PAGE_DOWN)
 
     #self.driver.execute_script("window.scrollTo(0, 1000)")