@@ -1,3 +1,5 @@
+import os
from helper import save_screenshot
-save_screenshot("test")
+screenshot_address = os.getenv("SCREENSHOT_PATH")
+save_screenshot(screenshot_address)