diff --git a/crystal_toolkit/components/structure.py b/crystal_toolkit/components/structure.py
index fbeaad9e..84f4a7b4 100644
--- a/crystal_toolkit/components/structure.py
+++ b/crystal_toolkit/components/structure.py
@@ -489,7 +489,7 @@ def download_image(image_data_timestamp, image_data, data):
             Output(self.id("download-structure"), "data"),
             Input(self.id("scene"), "fileTimestamp"),
             State(self.id("scene"), "fileType"),
-            State(self.id(), "data"),
+            State(self.id("graph"), "data"),
         )
         def download_structure(file_timestamp, download_option, data):
             if not file_timestamp: