diff --git a/scrapegraphai/graphs/screenshot_scraper_graph.py b/scrapegraphai/graphs/screenshot_scraper_graph.py
index 13046d93..f3ce608d 100644
--- a/scrapegraphai/graphs/screenshot_scraper_graph.py
+++ b/scrapegraphai/graphs/screenshot_scraper_graph.py
@@ -48,8 +48,8 @@ class ScreenshotScraperGraph(AbstractGraph):
             }
         )
         generate_answer_from_image_node = GenerateAnswerFromImageNode(
-            input="imgs",
-            output=["screenshots"],
+            input="screenshots",
+            output=["answer"],
             node_config={
                 "config": self.config
             }