diff --git a/test/test.py b/test/test.py
index 0f57190..0dac64b 100644
--- a/test/test.py
+++ b/test/test.py
@@ -36,7 +36,7 @@ def test_elements(self):
 
   def download_widget_image(self):
     self.driver.get("http://localhost:8383/voila/render/example.ipynb")
-    self.driver.set_window_size(1280, 1080)
+    self.driver.set_window_size(1280, 2000)
     time.sleep(3)
     self.driver.save_screenshot("widget-01.png")
 
@@ -45,6 +45,11 @@ def download_widget_image(self):
     time.sleep(3)
     self.driver.save_screenshot("widget-02.png")
 
+    #Move the end of the page
+    self.driver.find_element(By.TAG_NAME, 'body').send_keys(Keys.END)
+    time.sleep(3)
+    self.driver.save_screenshot("widget-03.png")
+
 test = test_widget()
 test.setup_method('Chrome')
 test.download_widget_image()